marvin/CNN_Website
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

changed server config

Browse Source
This commit is contained in:
marvin
2026-05-04 23:21:38 +02:00
parent 5137bddebd
commit dc9351a493
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
bird_cnn/server.py
+5 -1
View File
@@ -41,7 +41,11 @@ model.load_state_dict(torch.load(full_path, map_location=torch.device(device)))
model.to(device) model.to(device)
model.eval() 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") @app.post("/predict")
async def predict(file: UploadFile = File(...)): async def predict(file: UploadFile = File(...)):