lanzhiwang commited on
Commit
79a9c3d
1 Parent(s): d089c9a
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,4 +1,4 @@
1
- from transformers import ViTFeatureExtractor, BertTokenizer, VisionEncoderDecoderModel, AutoTokenizer
2
  import gradio as gr
3
 
4
  model=VisionEncoderDecoderModel.from_pretrained("priyank-m/vit-bert-OCR")
@@ -14,4 +14,4 @@ def run_ocr(image):
14
 
15
 
16
  demo = gr.Interface(fn=run_ocr, inputs="image", outputs="text")
17
- demo.launch()
 
1
+ from transformers import ViTFeatureExtractor, VisionEncoderDecoderModel, AutoTokenizer
2
  import gradio as gr
3
 
4
  model=VisionEncoderDecoderModel.from_pretrained("priyank-m/vit-bert-OCR")
 
14
 
15
 
16
  demo = gr.Interface(fn=run_ocr, inputs="image", outputs="text")
17
+ demo.launch()