Upload image_processing_gpu.py

image_processing_gpu.py

@@ -24,7 +24,8 @@ from utils import (
 # Variables globales pour OCR
 processor = None
 model = None
-OCR_MODEL_NAME = "TrOCR-base-handwritten"
+#OCR_MODEL_NAME = "TrOCR-base-handwritten"
+OCR_MODEL_NAME = "hoololi/trocr-base-handwritten-calctrainer"
 
 def init_ocr_model() -> bool:
     """Initialise TrOCR pour ZeroGPU"""
@@ -33,8 +34,10 @@ def init_ocr_model() -> bool:
     try:
         print("🔄 Chargement TrOCR (ZeroGPU)...")
         
-        processor = TrOCRProcessor.from_pretrained('microsoft/trocr-base-handwritten')
-        model = VisionEncoderDecoderModel.from_pretrained('microsoft/trocr-base-handwritten')
+        #processor = TrOCRProcessor.from_pretrained('microsoft/trocr-base-handwritten')
+        #model = VisionEncoderDecoderModel.from_pretrained('microsoft/trocr-base-handwritten')
+        processor = TrOCRProcessor.from_pretrained('hoololi/trocr-base-handwritten-calctrainer')
+        model = VisionEncoderDecoderModel.from_pretrained('hoololi/trocr-base-handwritten-calctrainer')
         
         # Optimisations
         model.eval()