Update app.py

app.py

@@ -19,6 +19,7 @@ from typing import Tuple
 #from transformers import AutoTokenizer, AutoModelForCausalLM
 import paramiko
 import gc
+import time
 #os.system("chmod +x ./cusparselt.sh")
 #os.system("./cusparselt.sh")
 #os.system("chmod +x ./cudnn.sh")
@@ -167,19 +168,23 @@ def set_gpu_duration(duration_choice):
 
 @spaces.GPU(duration=45)
 def generate_45s(*args, **kwargs):
-    return generate(*args, **kwargs)  # Call the common generate function
+    progress = kwargs.pop('progress', gr.Progress())  
+    return generate(*args, **kwargs, progress=progress)  # Pass progress to generate
 
 @spaces.GPU(duration=60)
 def generate_60s(*args, **kwargs):
-    return generate(*args, **kwargs)
+    progress = kwargs.pop('progress', gr.Progress())  
+    return generate(*args, **kwargs, progress=progress)  # Pass progress to generate
 
 @spaces.GPU(duration=80)
 def generate_80s(*args, **kwargs):
-    return generate(*args, **kwargs)
+    progress = kwargs.pop('progress', gr.Progress())  
+    return generate(*args, **kwargs, progress=progress)  # Pass progress to generate
 
 @spaces.GPU(duration=100)
 def generate_100s(*args, **kwargs):
-    return generate(*args, **kwargs)
+    progress = kwargs.pop('progress', gr.Progress())  
+    return generate(*args, **kwargs, progress=progress)  # Pass progress to generate
 
 def generate(
     model_choice: str,
@@ -195,8 +200,9 @@ def generate(
     randomize_seed: bool = False,
     use_resolution_binning: bool = True, 
     num_images: int = 1,  
-    progress=gr.Progress(track_tqdm=True),
+    progress=gr.Progress()  # Add progress as a keyword argument
 ):
+    progress(0.5, desc="Generating...") # Example usage
     global models
     pipe = models[model_choice]
     seed = int(randomize_seed_fn(seed, randomize_seed))