mirror of
https://github.com/RapidAI/RapidOCR.git
synced 2026-11-10 21:10:55 +08:00
feat(rapidocr_api): try import RapidOCR (#282)
try import RapidOCR from rapidocr_onnxruntime or rapidocr_paddle or rapidocr_openvino
This commit is contained in:
1 parent
a58d5d2f6e
commit
43efc5d55b
1 file changed
+12
-1
@@ -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))
|
||||
|
||||
|
||||
Reference in new issue
Block a user