Update app.py

app.py

@@ -37,12 +37,12 @@ feature_extractor = CLIPFeatureExtractor.from_pretrained("openai/clip-vit-base-p
 
 # Function 
 @spaces.GPU(duration=30,queue=False)
-def generate_image(prompt, base="Realistic", motion="", step=8, resolution="Square", duration=5, progress=gr.Progress()):
+def generate_image(prompt, base="Realistic", motion="", step=8, resolution="Square", progress=gr.Progress()):
     global step_loaded
     global base_loaded
     global motion_loaded
 
-    print(prompt, base, step, resolution, duration)
+    print(prompt, base, step, resolution)
 
     # Set resolution
     if resolution == "Square":
@@ -71,12 +71,11 @@ def generate_image(prompt, base="Realistic", motion="", step=8, resolution="Squa
     def progress_callback(i, t, z):
         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)
+    output = pipe(prompt=prompt, guidance_scale=1.2, num_inference_steps=step, width=width, height=height, callback=progress_callback, callback_steps=1)
 
     name = str(uuid.uuid4()).replace("-", "")
     path = f"/tmp/{name}.mp4"
-    export_to_video(output.frames, path, fps=10)  # Export all frames to video
+    export_to_video(output.frames[0], path, fps=10)
     return path
 
 
@@ -138,14 +137,6 @@ with gr.Blocks(css="style.css") as demo:
                 value="Square",
                 interactive=True
             )
-            select_duration = gr.Slider(
-                label='Video Duration (seconds)',
-                minimum=1,
-                maximum=30,
-                step=1,
-                value=5,
-                interactive=True
-            )
             submit = gr.Button(
                 scale=1,
                 variant='primary'
@@ -164,7 +155,7 @@ with gr.Blocks(css="style.css") as demo:
             prompt.submit
         ],
         fn=generate_image,
-        inputs=[prompt, select_base, select_motion, select_step, select_resolution, select_duration],
+        inputs=[prompt, select_base, select_motion, select_step, select_resolution],
         outputs=[video],
         api_name="instant_video",
         queue=False