Update app.py

app.py

@@ -85,14 +85,16 @@ def video_to_image(selected):
 @spaces.GPU(duration=120)
 def generate(
     media,
+    selected,
     seed: Optional[int] = -1,
+    num_inference_steps: int = 10,
+    animatediff_batch_size: int = 32,
+    animatediff_stride: int = 16,
     motion_bucket_id: int = 127,
     fps_id: int = 25,
-    num_inference_steps: int = 10,
     num_frames: int = 50,
-    output_folder: str = "outputs",
-    selected,
     prompt,
+    output_folder: str = "outputs",
     progress=gr.Progress(track_tqdm=True)):
 
     print(media)
@@ -133,11 +135,11 @@ def generate(
             cfg_scale=3, 
             clip_skip=2,
             controlnet_frames=input_video, num_frames=len(input_video),
-            num_inference_steps=10, 
+            num_inference_steps=num_inference_steps, 
             height=1024, 
             width=1024,
-            animatediff_batch_size=32, 
-            animatediff_stride=16,
+            animatediff_batch_size=animatediff_batch_size, 
+            animatediff_stride=animatediff_stride,
             vram_limit_level=0,
         )
         video = smoother(video)