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('cuda')
+    model = get_model(config)
     model = load(model, 'workdir/train-abinet/best-train-abinet.pth')
     charset = CharsetMapper(filename=config.dataset_charset_path, max_length=config.dataset_max_length + 1)
     

requirements.txt

@@ -14,4 +14,5 @@ matplotlib
 tqdm
 shapely
 scipy
-networkx
+networkx
+onnxruntime-gpu