Hugging Face's logo

Spaces:
Mageia
/
GOT-OCR-API
Sleeping

App Files Community
Mageia commited on
Commit
7b83985
·
unverified ·
1 Parent(s): dcec5ef

fix: model_name

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -24,7 +24,7 @@ app.add_middleware(
24
  )
25
 
26
  # 初始化模型
27
- model_name = "ucaslcl/GOT-OCR2_0"
28
  device = "cuda" if torch.cuda.is_available() else "cpu"
29
 
30
  tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
 
24
  )
25
 
26
  # 初始化模型
27
+ model_name = "Mageia/GOT-OCR2_0"
28
  device = "cuda" if torch.cuda.is_available() else "cpu"
29
 
30
  tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)