From dc9351a493c90454a0f24c420c30cba8c76086c5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Marvin=20Krau=C3=9Fer?= <koro.no.omo@gmail.com>
Date: Mon, 4 May 2026 23:21:38 +0200
Subject: [PATCH] changed server config

---
 bird_cnn/server.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/bird_cnn/server.py b/bird_cnn/server.py
index 20cea3c..76f787d 100644
--- a/bird_cnn/server.py
+++ b/bird_cnn/server.py
@@ -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(...)):