Update app.py

app.py

@@ -36,7 +36,12 @@ def perform_ocr(img, language):
     if language in ["English", "Both"]:
         # Ensure that inference is done on the correct device (GPU or CPU)
         with torch.no_grad():
-            res_eng = model_eng.chat(tokenizer_eng, img_path, ocr_type='ocr')
+            # Move inputs to the appropriate device
+            try:
+                res_eng = model_eng.chat(tokenizer_eng, img_path, ocr_type='ocr')
+            except Exception as e:
+                print(f"Error during English OCR: {e}")
+                res_eng = "Error during English OCR"
 
     if language in ["Hindi", "Both"]:
         img_cv = cv2.imread(img_path)