Update app.py

app.py

@@ -112,8 +112,6 @@ def generate(
     seed: int = -1,
     num_inference_steps: int = 10,
     num_frames: int = 30,
-    shape_height: int = 512,
-    shape_width: int = 512,
     height: int = 512,
     width: int = 512,
     animatediff_batch_size: int = 32,
@@ -136,7 +134,7 @@ def generate(
     if selected == "ExVideo" and image_in:
         image = Image.open(image_in)
         video = pipe(
-            input_image=image.resize((shape_height, shape_width)), 
+            input_image=image.resize((width, height)), 
             num_frames=num_frames, 
             fps=fps_id, 
             height=height, 
@@ -151,7 +149,7 @@ def generate(
     elif selected == "Diffutoon" and video_in:
         up_video = VideoData(
             video_file=video_in,
-            height=shape_height, width=shape_width)
+            height=height, width=width)
         input_video = [up_video[i] for i in range(1, num_frames)]
 
         video = pipe2(
@@ -192,7 +190,7 @@ with gr.Blocks(css=CSS, js=JS, theme="soft") as demo:
     gr.HTML("<h1><center>Exvideo📽️Diffutoon</center></h1>")
     gr.HTML("""
         <p><center>Exvideo and Diffutoon video generation
-        <br><b>Update</b>: Origin and Output resize, Frames control.
+        <br><b>Update</b>: Output resize, Frames length control.
         <br><b>Note</b>: ZeroGPU limited, Set the parameters appropriately.</center></p>
         """)
     with gr.Row():
@@ -228,33 +226,16 @@ with gr.Blocks(css=CSS, js=JS, theme="soft") as demo:
                 minimum=1, 
                 maximum=128,
             )
-            with gr.Row():
-                shape_height = gr.Slider(
-                    label="Shape Height",
-                    info="Resize Height",
-                    step=8,
-                    value=512, 
-                    minimum=256, 
-                    maximum=2560,
-                    )
-                shape_width = gr.Slider(
-                    label="Shape Width", 
-                    info="Resize Width",
-                    step=8,
-                    value=512, 
-                    minimum=256, 
-                    maximum=2560,
-                    )
             with gr.Row():
                 height = gr.Slider(
-                    label="Output Height", 
+                    label="Height", 
                     step=8,
                     value=512, 
                     minimum=256, 
                     maximum=2560,
                     )
                 width = gr.Slider(
-                    label="Output Width", 
+                    label="Width", 
                     step=8,
                     value=512, 
                     minimum=256, 
@@ -308,7 +289,7 @@ with gr.Blocks(css=CSS, js=JS, theme="soft") as demo:
     )
     selected.change(change_media, inputs=[image_in, video_in, selected], outputs=[image_in, video_in, prompt])
     video_in.upload(update_frames, inputs=[video_in], outputs=[num_frames])
-    submit_event = submit_btn.click(fn=generate, inputs=[video_in, image_in, selected, prompt, seed, num_inference_steps, num_frames, shape_height, shape_width, height, width, animatediff_batch_size, animatediff_stride, motion_bucket_id, fps_id], outputs=[video, seed], api_name="video")
+    submit_event = submit_btn.click(fn=generate, inputs=[video_in, image_in, selected, prompt, seed, num_inference_steps, num_frames, height, width, animatediff_batch_size, animatediff_stride, motion_bucket_id, fps_id], outputs=[video, seed], api_name="video")
     #stop_btn.click(fn=None, inputs=None, outputs=None, cancels=[submit_event])
     
 demo.queue().launch()