Hugging Face's logo

Spaces:
aarishshahmohsin
/
ocr_reader
Running

App Files Community
aarishshahmohsin commited on
Commit
1625392
•
1 Parent(s): 5aa95a6

aarishshahmohisn

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ rec_model.to(device)
32
  # got_model = AutoModel.from_pretrained('RufusRubin777/GOT-OCR2_0_CPU', trust_remote_code=True, low_cpu_mem_usage=True, device_map='cpu', use_safetensors=True)
33
  tokenizer = AutoTokenizer.from_pretrained('aarishshahmohsin/got_ocr_2', trust_remote_code=True, device_map='cpu')
34
  got_model = AutoModel.from_pretrained('aarishshahmohsin/got_ocr_2', trust_remote_code=True, low_cpu_mem_usage=True, device_map='cpu', use_safetensors=True)
35
- got_model = got_model.to_empty(device)
36
  got_model = got_model.eval().to(device)
37
  # return tokenizer, model
38
 
 
32
  # got_model = AutoModel.from_pretrained('RufusRubin777/GOT-OCR2_0_CPU', trust_remote_code=True, low_cpu_mem_usage=True, device_map='cpu', use_safetensors=True)
33
  tokenizer = AutoTokenizer.from_pretrained('aarishshahmohsin/got_ocr_2', trust_remote_code=True, device_map='cpu')
34
  got_model = AutoModel.from_pretrained('aarishshahmohsin/got_ocr_2', trust_remote_code=True, low_cpu_mem_usage=True, device_map='cpu', use_safetensors=True)
35
+ # got_model = got_model.to_empty()
36
  got_model = got_model.eval().to(device)
37
  # return tokenizer, model
38