Update app.py

app.py

@@ -43,12 +43,11 @@ def init_predictor():
         )
     )
     
-@spaces.GPU(duration=90)
-def generate_video(prompt, seed, image, size, steps, frames, progress=gr.Progress(track_tqdm=True) ):
+@spaces.GPU(duration=120)
+def generate_video(prompt, image, size, steps, frames, guidance_scale, progress=gr.Progress(track_tqdm=True) ):
     print(f"image:{type(image)}")
-    if seed == -1:
-        random.seed(time.time())
-        seed = int(random.randrange(4294967294))
+    random.seed(time.time())
+    seed = int(random.randrange(4294967294))
     kwargs = {
         "prompt": prompt,
         "height": size,
@@ -56,7 +55,7 @@ def generate_video(prompt, seed, image, size, steps, frames, progress=gr.Progres
         "num_frames": frames,
         "num_inference_steps": steps,
         "seed": seed,
-        "guidance_scale": 6.0,
+        "guidance_scale": guidance_scale,
         "embedded_guidance_scale": 1.0,
         "negative_prompt": "Aerial view, aerial view, overexposed, low quality, deformation, a poor composition, bad hands, bad teeth, bad eyes, bad limbs, distortion",
         "cfg_for": False,
@@ -76,7 +75,6 @@ with gr.Blocks() as demo:
             with gr.Row():
                 image = gr.Image(label="Upload Image", type="filepath")
                 prompt = gr.Textbox(label="Input Prompt")
-                seed = gr.Number(label="Random Seed", value=-1)
                 size = gr.Slider(
                     label="Size",
                     minimum=256,
@@ -93,16 +91,23 @@ with gr.Blocks() as demo:
                 )
                 steps = gr.Slider(
                     label="Number of Steps",
-                    minimum=16,
+                    minimum=1,
                     maximum=96,
                     step=1,
                     value=20,
                 )
+                guidance_scale = gr.Slider(
+                    label="Guidance Scale",
+                    minimum=1.0,
+                    maximum=16.0,
+                    step=.1,
+                    value=6.0,
+                )
             submit_button = gr.Button("Generate Video")
             output_video = gr.Video(label="Generated Video")
             submit_button.click(
                 fn=generate_video,
-                inputs=[prompt, seed, image, size, steps, frames],
+                inputs=[prompt, image, size, steps, frames, guidance_scale],
                 outputs=[output_video],
             )