Update app.py

app.py

@@ -153,9 +153,9 @@ cache_pipeline = {
 }
 
 
-@spaces.GPU
+@spaces.GPU(duration=120)
 def infer(
-    base_model, variant, prompt, num_inference_steps=4, seed=0, randomize_seed=True,
+    base_model, variant, prompt, num_inference_steps=4, seed=0, randomize_seed=True, progress=gr.Progress(track_tqdm=True),
 ):
     # if pipe_dict[base_model][variant] is None:
     #     if base_model == "ModelScope T2V":