Update app.py

app.py

@@ -63,7 +63,7 @@ def process_image(image, should_convert=False):
     # Generate OCR output
     text_prompt = ocr_processor.apply_chat_template(messages, add_generation_prompt=True)
     inputs = ocr_processor(text=[text_prompt], images=[image], padding=True, return_tensors="pt")
-    inputs = inputs.to("cuda")  # Use CPU if GPU is unavailable
+    inputs = inputs.to("cpu")  # Use CPU if GPU is unavailable
     output_ids = ocr_model.generate(**inputs, max_new_tokens=1024)
     output_text = ocr_processor.batch_decode(output_ids, skip_special_tokens=True, clean_up_tokenization_spaces=True)[0]