Solo448 commited on
Commit
03f6be5
·
verified ·
1 Parent(s): 8e7208b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ import shutil
15
  from pathlib import Path
16
  import tiktoken
17
  import verovio
18
- model_name = "stepfun-ai/GOT-OCR2_0" #ucaslcl/GOT-OCR2_0
19
 
20
  tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
21
  #model = AutoModel.from_pretrained(model_name, trust_remote_code=True, low_cpu_mem_usage=True, device_map='cuda', use_safetensors=True).eval().cuda()
 
15
  from pathlib import Path
16
  import tiktoken
17
  import verovio
18
+ model_name = "srimanth-d/GOT_CPU" #ucaslcl/GOT-OCR2_0
19
 
20
  tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
21
  #model = AutoModel.from_pretrained(model_name, trust_remote_code=True, low_cpu_mem_usage=True, device_map='cuda', use_safetensors=True).eval().cuda()