Update app.py

app.py

@@ -32,17 +32,17 @@ def init_predictor():
     )
     
 @spaces.GPU(duration=80)
-def generate_video(prompt, seed, image=None):
+def generate_video(prompt, seed, image=None, size, steps, frames):
     print(f"image:{type(image)}")
     if seed == -1:
         random.seed(time.time())
         seed = int(random.randrange(4294967294))
     kwargs = {
         "prompt": prompt,
-        "height": 320,
-        "width": 320,
-        "num_frames": 64,
-        "num_inference_steps": 10,
+        "height": size,
+        "width": size,
+        "num_frames": frames,
+        "num_inference_steps": steps,
         "seed": seed,
         "guidance_scale": 6.0,
         "embedded_guidance_scale": 1.0,
@@ -50,12 +50,13 @@ def generate_video(prompt, seed, image=None):
         "cfg_for": False,
     }
     assert image is not None, "please input image"
-    kwargs["image"] = load_image(image=image).resize((320,320), Image.LANCZOS)
+    img = load_image(image=image)
+    img.resize((size,size), Image.LANCZOS)
+    kwargs["image"] = img
     #global predictor
     output = predictor.inference(kwargs)
-    save_dir = f"./result"
-    os.makedirs(save_dir, exist_ok=True)
-    video_out_file = f"{save_dir}/{prompt[:100].replace('/','')}_{seed}.mp4"
+    save_dir = f"./"
+    video_out_file = f"{seed}.mp4"
     print(f"generate video, local path: {video_out_file}")
     export_to_video(output, video_out_file, fps=24)
     return video_out_file
@@ -66,11 +67,32 @@ def create_gradio_interface():
                 image = gr.Image(label="Upload Image", type="filepath")
                 prompt = gr.Textbox(label="Input Prompt")
                 seed = gr.Number(label="Random Seed", value=-1)
+                width = gr.Slider(
+                    label="Size",
+                    minimum=256,
+                    maximum=1024,
+                    step=16,
+                    value=448,
+                )
+                frames = gr.Slider(
+                    label="Number of Frames",
+                    minimum=16,
+                    maximum=256,
+                    step=8,
+                    value=64,
+                )
+                steps =  = gr.Slider(
+                    label="Number of Steps",
+                    minimum=16,
+                    maximum=96,
+                    step=1,
+                    value=20,
+                )
             submit_button = gr.Button("Generate Video")
             output_video = gr.Video(label="Generated Video")
             submit_button.click(
                 fn=generate_video,
-                inputs=[prompt, seed, image],
+                inputs=[prompt, seed, image, size, steps, frames],
                 outputs=[output_video],
             )
         return demo