Update app.py

app.py

@@ -35,7 +35,7 @@ def load_models():
 
 model, tokenizer, pipe = load_models()
 
-@spaces.GPU
+@spaces.GPU(duration=100)
 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}."
@@ -66,7 +66,7 @@ Below is the simple caption of an image with text. Please deduce the detailed de
     
     return combined_caption, enhanced_caption
 
-@spaces.GPU
+@spaces.GPU(duration=100)
 def generate_image(enhanced_caption, seed, num_inference_steps, guidance_scale):
     """Generate image using LeX-Lumina"""
     generator = torch.Generator("cpu").manual_seed(seed) if seed != 0 else None