Hugging Face's logo

Spaces:
dimasdeffieux
/
explain_lang
Sleeping

App Files Community
dimasdeffieux commited on
Commit
c9c18db
·
verified ·
1 Parent(s): 4c288df

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,7 +1,7 @@
1
- from transformers import AutoModel, AutoTokenizer
2
 
3
- tokenizer = AutoTokenizer.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_code=True)
4
- model = AutoModel.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_code=True, low_cpu_mem_usage=True, device_map='cuda', use_safetensors=True, pad_token_id=tokenizer.eos_token_id)
5
  model = model.eval().cuda()
6
 
7
 
 
1
+ from transformers import TFAutoModel, TFAutoTokenizer
2
 
3
+ tokenizer = TFAutoTokenizer.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_code=True)
4
+ model = TFAutoModel.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_code=True, low_cpu_mem_usage=True, device_map='cuda', use_safetensors=True, pad_token_id=tokenizer.eos_token_id)
5
  model = model.eval().cuda()
6
 
7