Update app.py

app.py

@@ -37,18 +37,11 @@ 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", progress=gr.Progress()):
+def generate_image(prompt, base="Realistic", motion="", step=8, progress=gr.Progress()):
     global step_loaded
     global base_loaded
     global motion_loaded
-
-    print(prompt, base, step, resolution)
-
-    # Set resolution
-    if resolution == "Square":
-        width, height = 512, 512
-    elif resolution == "Horizontal":
-        width, height = 1280, 720
+    print(prompt, base, step)
 
     if step_loaded != step:
         repo = "ByteDance/AnimateDiff-Lightning"
@@ -71,7 +64,7 @@ def generate_image(prompt, base="Realistic", motion="", step=8, resolution="Squa
     def progress_callback(i, t, z):
         progress((i+1, step))
 
-    output = pipe(prompt=prompt, guidance_scale=1.2, num_inference_steps=step, width=width, height=height, callback=progress_callback, callback_steps=1)
+    output = pipe(prompt=prompt, guidance_scale=1.2, num_inference_steps=step, callback=progress_callback, callback_steps=1)
 
     name = str(uuid.uuid4()).replace("-", "")
     path = f"/tmp/{name}.mp4"
@@ -83,7 +76,7 @@ def generate_image(prompt, base="Realistic", motion="", step=8, resolution="Squa
 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(
@@ -128,15 +121,6 @@ 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'
@@ -149,16 +133,15 @@ 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, select_resolution],
-        outputs=[video],
-        api_name="instant_video",
-        queue=False
+    ],
+        fn = generate_image,
+        inputs = [prompt, select_base, select_motion, select_step],
+        outputs = [video],
+        api_name = "instant_video",
+        queue = False
     )
 
-demo.queue().launch()
+demo.queue().launch(share=True)