Update app.py

app.py

@@ -22,6 +22,7 @@ def load_model():
     return Qwen2VLForConditionalGeneration.from_pretrained(
         "Qwen/Qwen2-VL-2B-Instruct",
         torch_dtype=torch.float32,  # Use float32 for CPU
+        low_cpu_mem_usage=True
     )
 
 # Load the model
@@ -76,6 +77,9 @@ def ocr_image(image, query="Extract text from the image", keyword=""):
 
 # Gradio interface
 def process_image(image, keyword=""):
+    max_size = 1024
+    if max(image.size) > max_size:
+        image.thumbnail((max_size, max_size))
     return ocr_image(image, keyword=keyword)
 
 # Update the Gradio interface:
@@ -90,4 +94,4 @@ interface = gr.Interface(
 )
 
 # Launch Gradio interface in Colab
-interface.launch()
+interface.launch(server_timeout=300)