Update app.py

app.py

@@ -40,7 +40,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=40)
+@spaces.GPU(duration=50)
 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}."
@@ -73,11 +73,13 @@ Below is the simple caption of an image with text. Please deduce the detailed de
     
     return combined_caption, enhanced_caption
 
-@spaces.GPU(duration=50)
+@spaces.GPU(duration=60)
 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)
+
+    print(f"enhanced caption:\n{enhanced_caption}")
     
     generator = torch.Generator("cpu").manual_seed(seed) if seed != 0 else None
     
@@ -186,4 +188,4 @@ with gr.Blocks() as demo:
     )
 
 if __name__ == "__main__":
-    demo.queue().launch(debug=True)
+    demo.launch(debug=True)