Update app.py

app.py

@@ -15,7 +15,8 @@ import shutil
 from pathlib import Path
 import tiktoken
 import verovio
-model_name = "ucaslcl/GOT-OCR2_0"
+model_name = "stepfun-ai/GOT-OCR2_0" #ucaslcl/GOT-OCR2_0
+
 tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
 #model = AutoModel.from_pretrained(model_name, trust_remote_code=True, low_cpu_mem_usage=True, device_map='cuda', use_safetensors=True).eval().cuda()
 model = AutoModel.from_pretrained(model_name, trust_remote_code=True, low_cpu_mem_usage=True, device_map='cpu', use_safetensors=True).eval()