Spaces:
Sleeping
Sleeping
File size: 1,150 Bytes
aa4728b 08606b0 1cdc160 08606b0 5bf1b3e 08606b0 1cdc160 |
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 44 45 46 47 |
from fastapi import FastAPI, File, UploadFile
app = FastAPI()
from rapidocr_onnxruntime import RapidOCR
# det_model_path同理
model = RapidOCR()
# 1、导入对应的包
from fastapi.middleware.cors import CORSMiddleware
app = FastAPI()
# 2、声明一个 源 列表；重点：要包含跨域的客户端 源
origins = [
"https://hycjack-fastapi-rapidocr.hf.space/",
"http://localhost",
"http://localhost:7860",
# 客户端的源
"http://127.0.0.1:7860"
]
# 3、配置 CORSMiddleware
app.add_middleware(
CORSMiddleware,
allow_origins=origins, # 允许访问的源
allow_credentials=True, # 支持 cookie
allow_methods=["*"], # 允许使用的请求方法
allow_headers=["*"] # 允许携带的 Headers
)
@app.get("/")
def read_root():
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
|