Update app.py

app.py

@@ -41,7 +41,7 @@ def truncate_caption_by_tokens(caption, max_tokens=256):
     return caption
 
 @spaces.GPU(duration=200)
-def generate_enhanced_caption(image_caption, text_caption, progress=gr.Progress(track_tqdm=True)):
+def generate_enhanced_caption(image_caption, text_caption):
     """Generate enhanced caption using the LeX-Enhancer model"""
     combined_caption = f"{image_caption}, with the text on it: {text_caption}."
     instruction = """
@@ -74,7 +74,7 @@ Below is the simple caption of an image with text. Please deduce the detailed de
     return combined_caption, enhanced_caption
 
 @spaces.GPU(duration=200)
-def generate_image(enhanced_caption, seed, num_inference_steps, guidance_scale, progress=gr.Progress(track_tqdm=True)):
+def generate_image(enhanced_caption, seed, num_inference_steps, guidance_scale):
     """Generate image using LeX-Lumina"""
     # Truncate the caption if it's too long
     enhanced_caption = truncate_caption_by_tokens(enhanced_caption, max_tokens=256)