ianluo commited on
Commit
eca2c4f
1 Parent(s): be95046
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -5,7 +5,7 @@ from transformers import pipeline
5
 
6
  app = FastAPI()
7
 
8
- pipe_ocr = pipeline(model="naver-clova-ocr/bros-base-uncased")
9
 
10
  app.mount("/", StaticFiles(directory="static", html=True), name="static")
11
 
@@ -16,4 +16,4 @@ def index() -> FileResponse:
16
  @app.get("/ocr")
17
  def ocr(input):
18
  result = pipe_ocr(input)
19
- print(result)
 
5
 
6
  app = FastAPI()
7
 
8
+ pipe_ocr = pipeline("object-detection", model="naver-clova-ocr/bros-base-uncased")
9
 
10
  app.mount("/", StaticFiles(directory="static", html=True), name="static")
11
 
 
16
  @app.get("/ocr")
17
  def ocr(input):
18
  result = pipe_ocr(input)
19
+ print(input)