Spaces:
Sleeping
Sleeping
File size: 1,081 Bytes
aa4728b 0125000 1cdc160 0125000 b9d651c 08606b0 eaff201 5bf1b3e eaff201 5bf1b3e 08606b0 1cdc160 b9d651c 1cdc160 b9d651c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
from fastapi import FastAPI, File, UploadFile
from PIL import Image
from fastapi.middleware.cors import CORSMiddleware
from rapidocr_onnxruntime import RapidOCR
import io
import numpy as np
import pandas as pd
model = RapidOCR()
app = FastAPI()
origins = [
"https://hycjack-fastapi-rapidocr.hf.space/",
"http://localhost",
"http://localhost:7860",
"http://127.0.0.1:7860"
]
app.add_middleware(
CORSMiddleware,
allow_origins=origins,
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"]
)
@app.get("/")
def read_root():
return {"Hello": "World!"}
@app.post("/ocr")
async def ocr(file: UploadFile = File(...)):
contents = await file.read()
image = Image.open(io.BytesIO(contents))
np_array = np.array(image)
ocr_result, elapse = model(np_array)
dt_boxes, rec_res, scores = list(zip(*ocr_result))
out_df = pd.DataFrame(
[[box, rec, score] for box, rec, score in zip(dt_boxes, rec_res, scores)],
columns=("box", "rec", "score"),
)
return out_df.to_dict(orient='records')
|