Hugging Face's logo

Spaces:
allnewthunder
/
ID_cards_v1
Sleeping

App Files Community
to-be commited on
Commit
c07d345
·
1 Parent(s): 4ef76f4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ def demo_process(input_img):
19
  task_prompt = f"<s_cord-v2>"
20
 
21
 
22
- pretrained_model = DonutModel.from_pretrained("to-be/ID_cards_v1")
23
  pretrained_model.eval()
24
 
25
  demo = gr.Interface(
 
19
  task_prompt = f"<s_cord-v2>"
20
 
21
 
22
+ pretrained_model = DonutModel.from_pretrained("to-be/ID_cards_v1",revision="main")
23
  pretrained_model.eval()
24
 
25
  demo = gr.Interface(