Update app/utils.py

app/utils.py

@@ -25,7 +25,7 @@ class OCRModel:
             try:
                 # First try with the standard approach
                 self.tokenizer = AutoTokenizer.from_pretrained(
-                    'stepfun-ai/GOT-OCR-2.0-hf',
+                    'stepfun-ai/GOT-OCR2_0',
                     trust_remote_code=True,
                     use_fast=False
                 )
@@ -33,12 +33,12 @@ class OCRModel:
                 logger.warning(f"Standard tokenizer failed, trying BertTokenizer: {str(e)}")
                 # Fall back to BertTokenizer if AutoTokenizer fails
                 self.tokenizer = BertTokenizer.from_pretrained(
-                    'stepfun-ai/GOT-OCR-2.0-hf',
+                    'stepfun-ai/GOT-OCR2_0',
                     trust_remote_code=True
                 )
                 
             self.model = AutoModel.from_pretrained(
-                'stepfun-ai/GOT-OCR-2.0-hf',
+                'stepfun-ai/GOT-OCR2_0',
                 trust_remote_code=True,
                 low_cpu_mem_usage=True,
                 device_map='auto',