zerogpu initialization

app.py

@@ -105,6 +105,7 @@ def get_negative_prompt(gender):
     else:  # Random
         return "(deformed iris, deformed pupils, semi-realistic, cgi, 3d, render, sketch, cartoon, drawing, anime, mutated hands and fingers:1.4), (deformed, distorted, disfigured:1.3), poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, disconnected limbs, mutation, mutated, ugly, disgusting, amputation"
 
+@spaces.GPU(duration=40)
 def generate_image(face_image_1, face_image_2, prompt, gender, width, height, num_inference_steps, seed):
     if seed is None:
         seed = int.from_bytes(os.urandom(4), "big")
@@ -116,7 +117,6 @@ def generate_image(face_image_1, face_image_2, prompt, gender, width, height, nu
     
     baby_image_path = morph_faces(face_image_1.name, face_image_2.name)
 
-    @spaces.GPU(duration=30)
     def generate_images(faceid_embeds, num_outputs=1):
         images = ip_model.generate(
             prompt=prompt,