Update app.py

app.py

@@ -37,7 +37,7 @@ feature_extractor = CLIPFeatureExtractor.from_pretrained("openai/clip-vit-base-p
 
 # Function 
 @spaces.GPU(duration=1,queue=False)
-def generate_image(prompt, base="Realistic", motion="", step=8, frame_multiplier=2, desired_fps=5, progress=gr.Progress()):
+def generate_image(prompt, base="Realistic", motion="", step=8, progress=gr.Progress()):
     global step_loaded
     global base_loaded
     global motion_loaded
@@ -64,21 +64,11 @@ def generate_image(prompt, base="Realistic", motion="", step=8, frame_multiplier
     def progress_callback(i, t, z):
         progress((i+1, step))
 
-    # Generate more frames by multiplying steps
-    output = pipe(
-        prompt=prompt,
-        guidance_scale=1.2,
-        num_inference_steps=step * frame_multiplier,  # More frames
-        callback=progress_callback,
-        callback_steps=1
-    )
+    output = pipe(prompt=prompt, guidance_scale=1.2, num_inference_steps=step, callback=progress_callback, callback_steps=1)
 
     name = str(uuid.uuid4()).replace("-", "")
     path = f"/tmp/{name}.mp4"
-    
-    # Export with lower FPS for longer video
-    export_to_video(output.frames, path, fps=desired_fps)
-    
+    export_to_video(output.frames[0], path, fps=10)
     return path