upload

README.md

@@ -1,6 +1,6 @@
 ---
 title: Japanese OCR
-emoji: 📈
+emoji: ⛩
 colorFrom: yellow
 colorTo: blue
 sdk: gradio

app.py

@@ -10,7 +10,11 @@ feature_extractor = AutoFeatureExtractor.from_pretrained(model_path)
 tokenizer = AutoTokenizer.from_pretrained(model_path)
 model = VisionEncoderDecoderModel.from_pretrained(model_path)
 
-
+examples = ['examples/01.png', 'examples/02.png', 'examples/03.png',
+            'examples/04.png', 'examples/05.png', 'examples/06.png',
+            'examples/07.png'
+            ]
+            
 def post_process(text):
     text = ''.join(text.split())
     text = text.replace('…', '...')
@@ -33,9 +37,10 @@ iface = gr.Interface(
     outputs="text",
     layout="horizontal",
     theme="huggingface",
-    title="Optical character recognition for Japanese text",
+    title="Optical Character Recognition for Japanese Text",
     description="A simple interface for OCR from Japanese manga",
     article= "Author: <a href=\"https://huggingface.co/vumichien\">Vu Minh Chien</a>. ",
     allow_flagging='never',
+    examples=examples
 )
-iface.launch(enable_queue=True, share=True, debug=True)
+iface.launch(enable_queue=True, cache_examples=True)

requirements.txt

@@ -1,3 +1,5 @@
 fugashi
 unidic_lite
 jaconv
+transformers
+torch