Update app.py

app.py

@@ -67,24 +67,19 @@ def generate_image(prompt, base="Realistic", motion="", step=8, resolution="Squa
             pipe.set_adapters(["motion"], [0.7])
         motion_loaded = motion
 
-    # Ensure num_frames aligns with the model's requirements
-    fps = 10
-    max_frames = 32  # Adjust based on model specs
-    num_frames = min((duration * fps) // max_frames * max_frames, max_frames)
-
     progress((0, step))
     def progress_callback(i, t, z):
-        progress((i + 1, step))
+        progress((i+1, step))
 
+    num_frames = duration * 10  # Assuming 10 FPS
     output = pipe(prompt=prompt, guidance_scale=1.2, num_inference_steps=step, width=width, height=height, num_frames=num_frames, callback=progress_callback, callback_steps=1)
 
     name = str(uuid.uuid4()).replace("-", "")
     path = f"/tmp/{name}.mp4"
-    export_to_video(output.frames, path, fps=fps)  # Export all frames to video
+    export_to_video(output.frames, path, fps=10)  # Export all frames to video
     return path
 
 
-
 # Gradio Interface
 with gr.Blocks(css="style.css") as demo:
     gr.HTML(