update app.py

app.py

@@ -34,7 +34,7 @@ textrec_model_list = ['ABINet', 'ASTER', 'CRNN', 'MASTER', 'NRTR', 'RobustScanne
 textkie_model_list = ['SDMGR','SDMGR']
 
 
-def ocr_inference(inputs, out_dir, det, det_weights, rec, rec_weights, device, batch_size):
+def ocr_inference(inputs, out_dir, det, det_weights, rec, rec_weights, device):
     init_args, call_args = parse_args()
     inputs = np.array(inputs)
     img_path = "demo_text_ocr.jpg"
@@ -52,7 +52,7 @@ def ocr_inference(inputs, out_dir, det, det_weights, rec, rec_weights, device, b
     
     call_args['inputs'] = img_path
     call_args['out_dir'] = out_dir
-    call_args['batch_size'] = int(batch_size)
+    call_args['batch_size'] = 1
     call_args['show'] = False
     call_args['save_pred'] = True
     call_args['save_vis'] = True
@@ -189,11 +189,11 @@ if __name__ == '__main__':
     output_image = gr.outputs.Image(type="pil", label="Output Image")
     output_json = gr.outputs.Textbox()
     download_test_image()
-    examples = [["demo_text_ocr.jpg", "results", "DBNet", None, "CRNN", None, "cpu", 1],
-                ["demo_text_det.jpg", "results", "FCENet", None, "ASTER", None, "cpu", 1],
-                ["demo_text_recog.jpg", "results", "PANet", None, "MASTER", None, "cpu", 1],
-                ["demo_densetext_det.jpg", "results", "PSENet", None, "CRNN", None, "cpu", 1],
-                ["demo_kie.jpg", "results", "TextSnake", None, "RobustScanner", None, "cpu", 1]
+    examples = [["demo_text_ocr.jpg", "results", "DBNet", None, "CRNN", None, "cpu"],
+                ["demo_text_det.jpg", "results", "FCENet", None, "ASTER", None, "cpu"],
+                ["demo_text_recog.jpg", "results", "PANet", None, "MASTER", None, "cpu"],
+                ["demo_densetext_det.jpg", "results", "PSENet", None, "CRNN", None, "cpu"],
+                ["demo_kie.jpg", "results", "TextSnake", None, "RobustScanner", None, "cpu"]
                 ]
 
     title = "MMOCR web demo"