Update app.py

app.py

@@ -1,3 +1,5 @@
+import os
+import spaces
 import gradio as gr
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
@@ -17,6 +19,7 @@ def load_models():
 
 model, tokenizer = load_models()
 
+@spaces.GPU(duration=70)
 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}."