ZienabM commited on
Commit
b052766
·
verified ·
1 Parent(s): 5c8c150

Update app/utils.py

Browse files
Files changed (1) hide show
  1. 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('ucaslcl/GOT-OCR2_0', trust_remote_code=True)
25
  self.model = AutoModel.from_pretrained(
26
- 'ucaslcl/GOT-OCR2_0',
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',