Update app.py

app.py

@@ -14,6 +14,8 @@ model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
 processor = AutoProcessor.from_pretrained(local_path)
 
 # Function to process image and text and generate the output
+@torch.inference_mode()
+@spaces.GPU(duration=120)  # Specify a duration to avoid timeout
 def generate_output(image, text, button_click):
     # Prepare input data
     messages = [