Update app.py

app.py

@@ -23,7 +23,7 @@ qwen_model = Qwen2VLForConditionalGeneration.from_pretrained(
 processor = AutoProcessor.from_pretrained("Qwen/Qwen2-VL-7B-Instruct", trust_remote_code=True)
 
 # Function for OCR and text extraction
-@spaces.GPU(duration=120)  # Increased GPU duration to 120 seconds
+# @spaces.GPU(duration=120)  # Increased GPU duration to 120 seconds
 def ocr_and_extract(image):
     try:
         # Save the uploaded image temporarily
@@ -85,6 +85,8 @@ def ocr_and_extract(image):
         traceback.print_exc()
         return f"Error: {error_message}"
 
+ocr_and_extract = spaces.GPU(duration=120)(ocr_and_extract)
+
 def search_keywords(extracted_text, keywords):
     if not extracted_text:
         return "No text extracted yet. Please upload an image."