Update app.py

app.py

@@ -10,6 +10,7 @@ from skyreelsinfer import TaskType
 from skyreelsinfer.skyreels_video_infer import SkyReelsVideoSingleGpuInfer
 from diffusers.utils import export_to_video
 from diffusers.utils import load_image
+from PIL import Image
 
 #predictor = None
 #task_type = None
@@ -40,7 +41,7 @@ def generate_video(prompt, seed, image=None):
         "prompt": prompt,
         "height": 320,
         "width": 320,
-        "num_frames": 32,
+        "num_frames": 64,
         "num_inference_steps": 10,
         "seed": seed,
         "guidance_scale": 6.0,
@@ -57,7 +58,7 @@ def generate_video(prompt, seed, image=None):
     video_out_file = f"{save_dir}/{prompt[:100].replace('/','')}_{seed}.mp4"
     print(f"generate video, local path: {video_out_file}")
     export_to_video(output, video_out_file, fps=24)
-    return video_out_file, kwargs
+    return video_out_file
 
 def create_gradio_interface():
         with gr.Blocks() as demo:
@@ -67,11 +68,10 @@ def create_gradio_interface():
                 seed = gr.Number(label="Random Seed", value=-1)
             submit_button = gr.Button("Generate Video")
             output_video = gr.Video(label="Generated Video")
-            output_params = gr.Textbox(label="Output Parameters")
             submit_button.click(
                 fn=generate_video,
                 inputs=[prompt, seed, image],
-                outputs=[output_video, output_params],
+                outputs=[output_video],
             )
         return demo