Update app.py

app.py

@@ -2,10 +2,9 @@ import gradio as gr
 from PIL import Image
 import numpy as np
 import cv2
-from transformers import TrOCRProcessor, VisionEncoderDecoderModel
 
-processor = TrOCRProcessor.from_pretrained("microsoft/trocr-base-handwritten")
-model = VisionEncoderDecoderModel.from_pretrained("microsoft/trocr-base-handwritten")
+from transformers import AutoModel
+model = AutoModel.from_pretrained("deepseek-ai/DeepSeek-OCR-2", trust_remote_code=True, dtype="auto")
 
 
 def segment_lines(image):