Spaces:
Sleeping
Sleeping
Update app/main.py
Browse files- 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:
|