Update app.py

app.py

@@ -59,11 +59,11 @@ if image_file is not None:
     if st.button("Run OCR"):
         # Use GOT-OCR2 model for plain text OCR (structured documents)
         with torch.no_grad():
-            res_plain = model.chat(tokenizer, temp_file_path, ocr_type='ocr')
+            res_plain = model.chat(tokenizer, temp_file_path, ocr_type='ocr')  # Removed device parameter
 
         # Perform formatted text OCR
         with torch.no_grad():
-            res_format = model.chat(tokenizer, temp_file_path, ocr_type='format')
+            res_format = model.chat(tokenizer, temp_file_path, ocr_type='format')  # Removed device parameter
 
         # Use EasyOCR for both English and Hindi text recognition
         result_easyocr = reader.readtext(temp_file_path, detail=0)
@@ -94,13 +94,13 @@ if image_file is not None:
 
         # Additional OCR types using GOT-OCR2
         with torch.no_grad():
-            res_fine_grained = model.chat(tokenizer, temp_file_path, ocr_type='ocr', ocr_box='')
+            res_fine_grained = model.chat(tokenizer, temp_file_path, ocr_type='ocr', ocr_box='')  # Removed device parameter
         st.subheader("Fine-Grained OCR Results:")
         st.write(res_fine_grained)
 
         # Render formatted OCR to HTML
         with torch.no_grad():
-            res_render = model.chat(tokenizer, temp_file_path, ocr_type='format', render=True, save_render_file='./demo.html')
+            res_render = model.chat(tokenizer, temp_file_path, ocr_type='format', render=True, save_render_file='./demo.html')  # Removed device parameter
         st.subheader("Rendered OCR Results (HTML):")
         st.write(res_render)