Update app/utils.py
Browse files- app/utils.py +2 -2
app/utils.py
CHANGED
@@ -21,9 +21,9 @@ class OCRModel:
|
|
21 |
logger.info("Initializing OCR model...")
|
22 |
|
23 |
# تهيئة النموذج والتوكينايزر
|
24 |
-
self.tokenizer = AutoTokenizer.from_pretrained('
|
25 |
self.model = AutoModel.from_pretrained(
|
26 |
-
'
|
27 |
trust_remote_code=True,
|
28 |
low_cpu_mem_usage=True,
|
29 |
device_map='cpu',
|
|
|
21 |
logger.info("Initializing OCR model...")
|
22 |
|
23 |
# تهيئة النموذج والتوكينايزر
|
24 |
+
self.tokenizer = AutoTokenizer.from_pretrained('RufusRubin777/GOT-OCR2_0_CPU', trust_remote_code=True)
|
25 |
self.model = AutoModel.from_pretrained(
|
26 |
+
'RufusRubin777/GOT-OCR2_0_CPU',
|
27 |
trust_remote_code=True,
|
28 |
low_cpu_mem_usage=True,
|
29 |
device_map='cpu',
|