Update app.py

app.py

@@ -37,11 +37,18 @@ 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, 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)
+
+    print(prompt, base, step, resolution)
+
+    # Set resolution
+    if resolution == "Square":
+        width, height = 512, 512
+    elif resolution == "Horizontal":
+        width, height = 1280, 720
 
     if step_loaded != step:
         repo = "ByteDance/AnimateDiff-Lightning"
@@ -64,7 +71,7 @@ def generate_image(prompt, base="Realistic", motion="", step=8, progress=gr.Prog
     def progress_callback(i, t, z):
         progress((i+1, step))
 
-    output = pipe(prompt=prompt, guidance_scale=1.2, num_inference_steps=step, 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"
@@ -76,7 +83,7 @@ def generate_image(prompt, base="Realistic", motion="", step=8, progress=gr.Prog
 with gr.Blocks(css="style.css") as demo:
     gr.HTML(
         "<h1><center>Textual Imagination : A Text To Video Synthesis</center></h1>"
-        )
+    )
     with gr.Group():
         with gr.Row():
             prompt = gr.Textbox(
@@ -121,6 +128,15 @@ with gr.Blocks(css="style.css") as demo:
                 value=4,
                 interactive=True
             )
+            select_resolution = gr.Dropdown(
+                label='Resolution',
+                choices=[
+                    "Square", 
+                    "Horizontal",
+                ],
+                value="Square",
+                interactive=True
+            )
             submit = gr.Button(
                 scale=1,
                 variant='primary'
@@ -133,17 +149,16 @@ with gr.Blocks(css="style.css") as demo:
         elem_id="video_output"
     )
 
-    gr.on(triggers=[
+    gr.on(
+        triggers=[
             submit.click,
             prompt.submit
-    ],
-        fn = generate_image,
-        inputs = [prompt, select_base, select_motion, select_step],
-        outputs = [video],
-        api_name = "instant_video",
-        queue = False
+        ],
+        fn=generate_image,
+        inputs=[prompt, select_base, select_motion, select_step, select_resolution],
+        outputs=[video],
+        api_name="instant_video",
+        queue=False
     )
 
 demo.queue().launch()
-
-Translate