Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -1,13 +1,13 @@
|
|
1 |
from transformers import ViTFeatureExtractor, BertTokenizer, VisionEncoderDecoderModel, AutoTokenizer, AutoFeatureExtractor
|
2 |
import gradio as gr
|
3 |
|
4 |
-
title="
|
5 |
-
description="
|
6 |
-
examples =[["demo_image/
|
7 |
|
8 |
|
9 |
|
10 |
-
model=VisionEncoderDecoderModel.from_pretrained("
|
11 |
tokenizer = AutoTokenizer.from_pretrained("xlm-roberta-base")
|
12 |
feature_extractor = AutoFeatureExtractor.from_pretrained("facebook/vit-mae-large")
|
13 |
|
|
|
1 |
from transformers import ViTFeatureExtractor, BertTokenizer, VisionEncoderDecoderModel, AutoTokenizer, AutoFeatureExtractor
|
2 |
import gradio as gr
|
3 |
|
4 |
+
title="Handwritten Phone Number OCR"
|
5 |
+
description="handwritten_phone_number_OCR is a Vision-Encoder-Decoder model (based on the concept of TrOCR) which uses pre-trained facebook's vit-mae-large as the encoder and xlm-roberta-base as the decoder. \nIt has been trained on MNIST (2M synthetic images)."
|
6 |
+
examples =[["demo_image/0313474611_lauf25.png"], ["demo_image/0466975865_nr4ywx.png"], ["demo_image/0473227403_7lnod1.png"], ["demo_image/0728880927_jr987p.png"], ["demo_image/0922853144_1o4ay5.png"]]
|
7 |
|
8 |
|
9 |
|
10 |
+
model=VisionEncoderDecoderModel.from_pretrained("dawars/handwritten_phone_number_OCR")
|
11 |
tokenizer = AutoTokenizer.from_pretrained("xlm-roberta-base")
|
12 |
feature_extractor = AutoFeatureExtractor.from_pretrained("facebook/vit-mae-large")
|
13 |
|