Update app.py

app.py

@@ -41,7 +41,7 @@ def truncate_caption_by_tokens(caption, max_tokens=256):
         print(f"Caption was truncated from {len(tokens)} tokens to {max_tokens} tokens")
     return caption
 
-@spaces.GPU(duration=50)
+@spaces.GPU(duration=70)
 def generate_enhanced_caption(image_caption, text_caption):
     # model.to("cuda")
     """Generate enhanced caption using the LeX-Enhancer model"""
@@ -107,7 +107,7 @@ def generate_image(enhanced_caption, seed, num_inference_steps, guidance_scale):
     
     return image
 
-@spaces.GPU(duration=100)
+@spaces.GPU(duration=200)
 def run_pipeline(image_caption, text_caption, seed, num_inference_steps, guidance_scale, enable_enhancer):
     """Run the complete pipeline from captions to final image"""
     combined_caption = f"{image_caption}, with the text on it: {text_caption}."