Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -15,7 +15,7 @@ import shutil
|
|
15 |
from pathlib import Path
|
16 |
import tiktoken
|
17 |
import verovio
|
18 |
-
model_name = "
|
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()
|