Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from transformers import AutoModel, AutoTokenizer
 from PIL import Image
-import numpy as np
+import tempfile
 
 # Load the tokenizer and model
 tokenizer = AutoTokenizer.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_code=True)
@@ -15,11 +15,12 @@ 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)
+        # Save the image to a temporary file
+        with tempfile.NamedTemporaryFile(suffix=".jpg", delete=False) as temp_file:
+            image.save(temp_file.name)
 
-        # Perform OCR using the model
-        res = model.chat(tokenizer, image_array, ocr_type='ocr')  # Adjusted to pass the image array
+            # Perform OCR using the model
+            res = model.chat(tokenizer, temp_file.name, ocr_type='ocr')  # Pass the file path
 
         return res
     except Exception as e: