Spaces:
Runtime error
Runtime error
Update pp_ocr.py
Browse files
pp_ocr.py
CHANGED
|
@@ -11,7 +11,10 @@ pp_ocrv3 = hub.Module(name=module_name)
|
|
| 11 |
|
| 12 |
def inference_img(img):
|
| 13 |
with tempfile.TemporaryDirectory() as tempdir_name:
|
| 14 |
-
|
|
|
|
|
|
|
|
|
|
| 15 |
result_names = os.listdir(tempdir_name)
|
| 16 |
result_image = Image.open(os.path.join(tempdir_name, result_names[0]))
|
| 17 |
return result_image
|
|
|
|
| 11 |
|
| 12 |
def inference_img(img):
|
| 13 |
with tempfile.TemporaryDirectory() as tempdir_name:
|
| 14 |
+
if module_name == "ch_pp-ocrv3":
|
| 15 |
+
pp_ocrv3.recognize_text(images=[img], use_gpu=False, output_dir=tempdir_name, visualization=True)
|
| 16 |
+
elif module_name == "multi_languages_ocr_db_crnn":
|
| 17 |
+
pp_ocrv3.recognize_text(images=[img], output_dir=tempdir_name, visualization=True)
|
| 18 |
result_names = os.listdir(tempdir_name)
|
| 19 |
result_image = Image.open(os.path.join(tempdir_name, result_names[0]))
|
| 20 |
return result_image
|