Update app.py

app.py

@@ -155,7 +155,7 @@ cache_pipeline = {
 
 @spaces.GPU
 def infer(
-    base_model, variant, prompt, seed=0, randomize_seed=True, num_inference_steps=6
+    base_model, variant, prompt, num_inference_steps=4, seed=0, randomize_seed=True,
 ):
     # if pipe_dict[base_model][variant] is None:
     #     if base_model == "ModelScope T2V":
@@ -246,42 +246,50 @@ examples = [
         "ModelScope T2V",
         "LAION-aes",
         "Aerial uhd 4k view. mid-air flight over fresh and clean mountain river at sunny summer morning. Green trees and sun rays on horizon. Direct on sun.",
+        4
     ],
     ["ModelScope T2V", "Anime", "Timelapse misty mountain landscape"],
     [
         "ModelScope T2V",
         "WebVid",
         "Back of woman in shorts going near pure creek in beautiful mountains.",
+        4
     ],
     [
         "ModelScope T2V",
         "3D Cartoon",
         "A rotating pandoro (a traditional italian sweet yeast bread, most popular around christmas and new year) being eaten in time-lapse.",
+        4
     ],
     [
         "ModelScope T2V",
         "Realistic",
         "Slow motion avocado with a stone falls and breaks into 2 parts with splashes",
+        4
     ],
     [
-        "AnimateDiff (SD1.5)",
+        "AnimateDiff (RealisticVision)",
         "LAION-aes",
         "Slow motion of delicious salmon sachimi set with green vegetables leaves served on wood plate. make homemade japanese food at home.-dan",
+        8
     ],
     [
-        "AnimateDiff (SD1.5)",
+        "AnimateDiff (RealisticVision)",
         "WebVid",
         "Blooming meadow panorama zoom-out shot heavenly clouds and upcoming thunderstorm in mountain range harz, germany.",
+        8
     ],
     [
         "AnimateDiff (RealisticVision)",
         "LAION-aes",
         "A young woman in a yellow sweater uses vr glasses, sitting on the shore of a pond on a background of dark waves. a strong wind develops her hair, the sun's rays are reflected from the water.",
+        8
     ],
     [
         "AnimateDiff (RealisticVision)",
         "LAION-aes",
         "Female running at sunset. healthy fitness concept",
+        8
     ],
 ]
 
@@ -391,7 +399,7 @@ with gr.Blocks(css=css) as demo:
 
         gr.Examples(
             examples=examples,
-            inputs=[base_model, variant_dropdown, prompt],
+            inputs=[base_model, variant_dropdown, prompt, num_inference_steps],
             cache_examples=True,
             fn=infer,
             outputs=[result],
@@ -403,9 +411,9 @@ with gr.Blocks(css=css) as demo:
             base_model,
             variant_dropdown,
             prompt,
+            num_inference_steps,
             seed,
             randomize_seed,
-            num_inference_steps,
         ],
         outputs=[result],
     )