First model version

app.py

@@ -22,7 +22,7 @@ import torch
 def infer(filepath):
     config = Config('configs/rec/train_abinet.yaml')
     config.model_vision_checkpoint = None
-    model = get_model(config).to_device('cuda')
+    model = get_model(config).to('cuda')
     model = load(model, 'workdir/train-abinet/best-train-abinet.pth')
     charset = CharsetMapper(filename=config.dataset_charset_path, max_length=config.dataset_max_length + 1)
     
@@ -58,6 +58,6 @@ iface = gr.Interface(
     description="毕设题目：自然场景中任意形状文字的检测与识别\n目前进度：检测",
     inputs=[gr.inputs.Image(label="image", type="filepath")],
     outputs=[gr.outputs.Image(), gr.outputs.Dataframe(headers=['word'])], #
-    #examples=['example1.jpg', 'example2.jpg', 'example3.jpg'],
+    examples=['figs/test/CANDY.png', 'figs/test/ESPLANADE.png', 'figs/test/KAPPA.png'],
     #article="<a href=\"https://github.com/MhLiao/MaskTextSpotterV3\">GitHub Repo</a>",
-).launch(enable_queue=True)
+).launch(enable_queue=True, cache_examples=True)