Update app.py

app.py

@@ -70,6 +70,8 @@ def get_duration(
     steps,
     seed,
     randomize_seed,
+    width,
+    height,
     progress,
 ):
     return steps * 2
@@ -83,6 +85,8 @@ def generate_image(
     steps = 27,
     seed = 42,
     randomize_seed = False,
+    width=1024, 
+    height=1024,
     progress=gr.Progress(track_tqdm=True),
 ):
     """
@@ -126,8 +130,8 @@ def generate_image(
     out_img = pipe(
         prompt=prompt,
         negative_prompt=negative_prompt,
-        height=1024,
-        width=1024,
+        height=height,
+        width=width,
         num_frames=1,
         guidance_scale=float(guidance_scale),
         guidance_scale_2=float(guidance_scale_2),
@@ -158,6 +162,9 @@ with gr.Blocks(css=css) as demo:
             negative_prompt_input = gr.Textbox(label="Negative Prompt", value=default_negative_prompt, lines=3)
             seed_input = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=42, interactive=True)
             randomize_seed_checkbox = gr.Checkbox(label="Randomize seed", value=True, interactive=True)
+            with gr.Row():
+                width = gr.Slider(label="Width", value=1024, minimum=64, maximum=1280, step=16)
+                height = gr.Slider(label="Height", value=1024, minimum=64, maximum=1280, step=16)
             steps_slider = gr.Slider(minimum=1, maximum=40, step=1, value=27, label="Inference Steps") 
             guidance_scale_input = gr.Slider(minimum=0.0, maximum=10.0, step=0.5, value=3.5, label="Guidance Scale - high noise stage")
             guidance_scale_2_input = gr.Slider(minimum=0.0, maximum=10.0, step=0.5, value=4, label="Guidance Scale 2 - low noise stage")