HycJack commited on
Commit
eaff201
1 Parent(s): b9d651c
Files changed (1) hide show
  1. main.py +5 -12
main.py CHANGED
@@ -5,27 +5,23 @@ from rapidocr_onnxruntime import RapidOCR
5
  import io
6
  import numpy as np
7
  import pandas as pd
8
- # det_model_path同理
9
- model = RapidOCR()
10
 
 
11
  app = FastAPI()
12
 
13
- # 2、声明一个 源 列表;重点:要包含跨域的客户端 源
14
  origins = [
15
  "https://hycjack-fastapi-rapidocr.hf.space/",
16
  "http://localhost",
17
  "http://localhost:7860",
18
- # 客户端的源
19
  "http://127.0.0.1:7860"
20
  ]
21
 
22
- # 3、配置 CORSMiddleware
23
  app.add_middleware(
24
  CORSMiddleware,
25
- allow_origins=origins, # 允许访问的源
26
- allow_credentials=True, # 支持 cookie
27
- allow_methods=["*"], # 允许使用的请求方法
28
- allow_headers=["*"] # 允许携带的 Headers
29
  )
30
 
31
  @app.get("/")
@@ -34,11 +30,8 @@ def read_root():
34
 
35
  @app.post("/ocr")
36
  async def ocr(file: UploadFile = File(...)):
37
- # 读取上传的文件内容
38
  contents = await file.read()
39
- # 使用Pillow打开图像
40
  image = Image.open(io.BytesIO(contents))
41
- # 将图像转换为numpy数组
42
  np_array = np.array(image)
43
  ocr_result, elapse = model(np_array)
44
  dt_boxes, rec_res, scores = list(zip(*ocr_result))
 
5
  import io
6
  import numpy as np
7
  import pandas as pd
 
 
8
 
9
+ model = RapidOCR()
10
  app = FastAPI()
11
 
 
12
  origins = [
13
  "https://hycjack-fastapi-rapidocr.hf.space/",
14
  "http://localhost",
15
  "http://localhost:7860",
 
16
  "http://127.0.0.1:7860"
17
  ]
18
 
 
19
  app.add_middleware(
20
  CORSMiddleware,
21
+ allow_origins=origins,
22
+ allow_credentials=True,
23
+ allow_methods=["*"],
24
+ allow_headers=["*"]
25
  )
26
 
27
  @app.get("/")
 
30
 
31
  @app.post("/ocr")
32
  async def ocr(file: UploadFile = File(...)):
 
33
  contents = await file.read()
 
34
  image = Image.open(io.BytesIO(contents))
 
35
  np_array = np.array(image)
36
  ocr_result, elapse = model(np_array)
37
  dt_boxes, rec_res, scores = list(zip(*ocr_result))