2nd draft

app.py

@@ -12,8 +12,8 @@ import gradio as gr
 import torch
 
 
-# from mmocr.apis import MMOCRInferencer
-# ocr = MMOCRInferencer(det="TextSnake", rec="ABINet_Vision")
+from mmocr.apis import MMOCRInferencer
+ocr = MMOCRInferencer(det="TextSnake", rec="ABINet_Vision")
 
 url = "https://upload.wikimedia.org/wikipedia/commons/thumb/5/5b/Draft_Marks_on_the_Bow_of_Kruzenshtern_Port_of_Tallinn_16_July_2011.jpg/1600px-Draft_Marks_on_the_Bow_of_Kruzenshtern_Port_of_Tallinn_16_July_2011.jpg"
 path_input = "./example1.jpg"
@@ -25,21 +25,15 @@ urllib.request.urlretrieve(url, filename=path_input)
 
 
 path_img_output_folder = "./demo-out"
-# if not os.path.exists(path_img_output_folder):
-#     os.makedirs(path_img_output_folder)
-
 path_img_input_folder = "./demo-input"
-# if not os.path.exists(path_img_input_folder):
-#     os.makedirs(path_img_input_folder)
-
 
 def do_process(img):
     img_name = "tmp.jpg"
     path_input = os.path.join(path_img_input_folder, img_name)
     path_output = os.path.join(path_img_output_folder, "vis", img_name)
     img.save(path_input)
-    img.save(path_output)
-    # result = ocr(path_input, out_dir=path_img_output_folder, save_vis=True)
+    # img.save(path_output)
+    result = ocr(path_input, out_dir=path_img_output_folder, save_vis=True)
     img_res = Image.open(path_output)
     return img_res