Update app.py

app.py

@@ -56,7 +56,6 @@ print_ckpt_structure()
 # =========================================
 
 from inference.flovd_demo import generate_video
-
 def run_inference(prompt, image, pose_type, speed, use_flow_integration, cam_pose_name):
     log_buffer = io.StringIO()
     sys_stdout = sys.stdout
@@ -67,6 +66,11 @@ def run_inference(prompt, image, pose_type, speed, use_flow_integration, cam_pos
         print("🚀 Starting inference...", flush=True)
         os.makedirs("input_images", exist_ok=True)
         image_path = "input_images/input_image.png"
+
+        # ✅ Convert NumPy to PIL if necessary
+        if not isinstance(image, Image.Image):
+            image = Image.fromarray(image.astype("uint8"))
+
         image.save(image_path)
         print(f"📸 Saved input image to {image_path}", flush=True)