First model version

app.py

@@ -19,7 +19,9 @@ from utils import Config, Logger, CharsetMapper
 import torch
 
 
-def infer(filepath):
+def infer(img):
+    filepath = './input.png'
+    img.save(filepath)
     config = Config('configs/rec/train_abinet.yaml')
     config.model_vision_checkpoint = None
     model = get_model(config)
@@ -53,7 +55,7 @@ def infer(filepath):
 
 blocks = gr.Blocks()
 
-input_image = gr.Image(label="image", type="filepath")
+input_image = gr.Image(label="image", type="pil")
 output_image = gr.Image(label="out_img", type="filepath")
 output_word = gr.Dataframe(label="out_word", headers=['word'])