Wootang01 commited on
Commit
92679b1
1 Parent(s): 6db8345

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ description = "Gradio Demo for DiT, the Document Image Transformer pre-trained o
6
  article = "<p style='text-align: center'><a href='https://huggingface.co/microsoft/dit-base-finetuned-rvlcdip' target='_blank'>Huggingface Model</a></p>"
7
 
8
  pipe = pipeline(task="image-classification",
9
- model="microsoft/dit-base-finetuned-rvlcdip")
10
  gr.Interface.from_pipeline(pipe,
11
  title=title,
12
  description=description,
 
6
  article = "<p style='text-align: center'><a href='https://huggingface.co/microsoft/dit-base-finetuned-rvlcdip' target='_blank'>Huggingface Model</a></p>"
7
 
8
  pipe = pipeline(task="image-classification",
9
+ model="microsoft/beit-base-patch16-224-pt22k-ft22k")
10
  gr.Interface.from_pipeline(pipe,
11
  title=title,
12
  description=description,