Hugging Face's logo

Spaces:
VIDraft
/
GOT-OCR-Transformers
Sleeping

App Files Community
yonigozlan HF Staff commited on
Commit
c61403c
·
verified ·
1 Parent(s): f67a5c0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ from render import render_ocr_text
18
  from transformers import AutoModelForImageTextToText, AutoProcessor
19
  from transformers.image_utils import load_image
20
 
21
- model_name = "yonigozlan/GOT-OCR-2.0-hf"
22
 
23
  device = "cuda" if torch.cuda.is_available() else "cpu"
24
 
 
18
  from transformers import AutoModelForImageTextToText, AutoProcessor
19
  from transformers.image_utils import load_image
20
 
21
+ model_name = "stepfun-ai/GOT-OCR-2.0-hf"
22
 
23
  device = "cuda" if torch.cuda.is_available() else "cpu"
24