add file

Dockerfile

@@ -8,4 +8,4 @@ RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
 COPY . .
 
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

main.py

@@ -1,7 +1,22 @@
 from fastapi import FastAPI
 
 app = FastAPI()
+from rapidocr_onnxruntime import RapidOCR
+
+# det_model_path同理
+model = RapidOCR()
 
 @app.get("/")
 def read_root():
-    return {"Hello": "World!"}
+    return {"Hello": "World!"}
+	
+@app.post("/ocr")	
+async def ocr(file: UploadFile = File(...)):
+    # 读取上传的文件内容
+    contents = await file.read()
+    # 使用Pillow打开图像
+    image = Image.open(io.BytesIO(contents))
+    # 将图像转换为numpy数组
+    np_array = np.array(image)
+    result, elapse = model(np_array)
+    return result

requirements.txt

@@ -4,4 +4,6 @@ sentencepiece==0.1.*
 torch==1.11.*
 transformers==4.*
 uvicorn[standard]==0.17.*
-
+Pillow
+onnxruntime
+rapidocr_onnxruntime