feat(rapidocr_api): try import RapidOCR (#282)

try import RapidOCR from rapidocr_onnxruntime or rapidocr_paddle or rapidocr_openvino
This commit is contained in:
huangbinghe authored and GitHub committed 2024-12-04 08:27:03 +08:00
1 parent a58d5d2f6e
commit 43efc5d55b
1 file changed
+12 -1
+12 -1
View File
@@ -9,12 +9,23 @@ import os
import sys
from pathlib import Path
from typing import Dict
import importlib.util
import numpy as np
import uvicorn
from fastapi import FastAPI, Form, UploadFile
from PIL import Image
from rapidocr_onnxruntime import RapidOCR
if importlib.util.find_spec("rapidocr_runtime"):
from rapidocr_onnxruntime import RapidOCR
elif importlib.util.find_spec("rapidocr_paddle"):
from rapidocr_paddle import RapidOCR
elif importlib.util.find_spec("rapidocr_openvino"):
from rapidocr_openvino import RapidOCR
else:
raise ImportError(
"Pleas install one of [rapidocr-runtime,rapidocr-paddle,rapidocr-openvino]"
)
sys.path.append(str(Path(__file__).resolve().parent.parent))