Update app.py

app.py

@@ -353,12 +353,12 @@ def spirit_animal_with_background(image_path, num_images = 4):
     return prompt, generated_images
 
 
-def generate_multiple_animals(image_path, keep_background=True, num_images = 4):
+def generate_multiple_animals(image_path, keep_background=True, num_images = 4, height_multiplier = 1.5):
 
     image = cv2.imread(image_path)
     image_rgb = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
 
-    image_rgb = crop_face_to_square(image_rgb)
+    image_rgb = crop_face_to_square(image_rgb,  height_multiplier = height_multiplier)
 
     original_image = Image.fromarray(image_rgb)
     original_width, original_height = original_image.size
@@ -514,7 +514,7 @@ def generate_spirit_animal_video(driving_video_path):
         print(f"First frame saved to: {first_frame_path}")
 
         # Generate spirit animal image
-        _, input_image = generate_multiple_animals(first_frame_path, True, 1)
+        _, input_image = generate_multiple_animals(first_frame_path, True, 1, height_multiplier = 1)
         if input_image is None or not input_image:
             print("Error: Spirit animal generation failed.")
             return None