ianluo commited on
Commit
5a9777a
1 Parent(s): eca2c4f
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("object-detection", 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(input)
 
5
 
6
  app = FastAPI()
7
 
8
+ pipe_ocr = pipeline("text0recognition", 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(result)