Update app.py

app.py

@@ -13,7 +13,6 @@ import tempfile
 import gc
 from PIL import Image
 import numpy as np
-from diffusers.utils import export_to_video
 
 def pil_to_tempfile(image):
     tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".png")
@@ -77,29 +76,21 @@ def video_tool(video_image_input: Image.Image) -> str:
     
     try:
         FPS = 16
-        num_frames = aligned_num_frames(1.5, FPS)
+        num_frames = aligned_num_frames(1, FPS)
     
-        frames = video_client.image_to_video(
+        video_bytes = video_client.image_to_video(
             image=video_image_input.resize((480, 480)),
             prompt="high quality",
             negative_prompt="low quality, deformed, grainy",
             num_frames=num_frames,
-            num_inference_steps=int(13),
-            guidance_scale=float(1),
+            num_inference_steps=int(20),
+            guidance_scale=float(1.4),
         )
 
-        np_frames = []
-        for f in frames:
-            if isinstance(f, Image.Image):
-                np_frames.append(np.array(f))
-            else:
-                np_frames.append(f)
-
-    
         out = tempfile.mktemp(suffix=".mp4")
-        export_to_video(np_frames, out, fps=FPS)
+        with open(out, "wb") as f:
+            f.write(video_bytes)
 
-        gc.collect()
         video_output = out
         
         return "Video successfully generated and stored for Gradio UI."