d4ndelion commited on
Commit
33ddf13
1 Parent(s): 8b031d2

Update app/main.py

Browse files
Files changed (1) hide show
  1. app/main.py +1 -6
app/main.py CHANGED
@@ -7,7 +7,7 @@ from PIL import Image
7
  from io import BytesIO
8
  from app.inferrer import Inferrer
9
 
10
- app = FastAPI()
11
 
12
  app.add_middleware(
13
  CORSMiddleware,
@@ -21,11 +21,6 @@ class_names = ["apples", "pears", "oranges"]
21
  recognizer = Inferrer("app/pear_apple_model.pth", len(class_names))
22
 
23
 
24
- @app.get("/", response_class=HTMLResponse)
25
- async def root(request: Request):
26
- return "OK"
27
-
28
-
29
  @app.post("/infer")
30
  async def predict(request: Request, file: UploadFile = File(...)):
31
  try:
 
7
  from io import BytesIO
8
  from app.inferrer import Inferrer
9
 
10
+ app = FastAPI(docs_url="/")
11
 
12
  app.add_middleware(
13
  CORSMiddleware,
 
21
  recognizer = Inferrer("app/pear_apple_model.pth", len(class_names))
22
 
23
 
 
 
 
 
 
24
  @app.post("/infer")
25
  async def predict(request: Request, file: UploadFile = File(...)):
26
  try: