Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from transformers import AutoModel, AutoTokenizer
 from PIL import Image
+import numpy as np
 
 # Load the tokenizer and model
 tokenizer = AutoTokenizer.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_code=True)
@@ -14,8 +15,11 @@ def perform_ocr(image):
         if image.mode != "RGB":
             image = image.convert("RGB")
 
+        # Convert the image to a format suitable for the model (if needed)
+        image_array = np.array(image)
+
         # Perform OCR using the model
-        res = model.chat(tokenizer, image, ocr_type='ocr')
+        res = model.chat(tokenizer, image_array, ocr_type='ocr')  # Adjusted to pass the image array
 
         return res
     except Exception as e: