changed server config

2026-05-04 23:21:38 +02:00
parent 5137bddebd
commit dc9351a493
1 changed files with 5 additions and 1 deletions
@@ -41,7 +41,11 @@ model.load_state_dict(torch.load(full_path, map_location=torch.device(device)))
 model.to(device)
 model.eval()

-app = FastAPI(root_path="/cnn-api")
+app = FastAPI(
+    root_path="/cnn-api",
+    openapi_url="/cnn-api/openapi.json",
+    docs_url="/cnn-api/docs"
+)

@app.post("/predict")
 async def predict(file: UploadFile = File(...)):