Update main.py

main.py

@@ -7,8 +7,8 @@ import torch
 
 device = 'gpu' if torch.cuda.is_available() else 'cpu'
 
-ocr = MMOCRInferencer(det_config='model/det/config.py',
-            det_ckpt='model/det/model.pth',
+ocr = MMOCRInferencer(det='model/det/config.py',
+            det_weights='model/det/model.pth',
             device=device)
 
 def get_rec(points):
@@ -27,7 +27,7 @@ def predict(image_input, score_threshold):
     print('image shape', image_input.shape, file=sys.stderr)
 
     try:
-        output = ocr.readtext(image_input)
+        return ocr.readtext(image_input)
     except:
         output = dict(det_polygons=[], det_scores=[])