Update app.py

app.py

@@ -41,6 +41,7 @@ def build_prompt(task="caption", input=None, sep="\n\n### "):
     return p
 
 # Define the function to generate text from the image and prompt
+@spaces.GPU(duration=120)
 def generate_text(image, task, input_text=None):
     prompt = build_prompt(task=task, input=input_text)
     inputs = processor(images=image, return_tensors="pt")