diff --git a/main.py b/main.py index e953071..48d45d7 100644 --- a/main.py +++ b/main.py @@ -58,4 +58,4 @@ def feedback_loop(data: List[FeedbackIn]): # Main function to start the app when main.py is called if __name__ == "__main__": # Uvicorn is used to run the server and listen for incoming API requests on 0.0.0.0:8888 - uvicorn.run("main:app", host="0.0.0.0", port=8887, reload=True) + uvicorn.run("main:app", host="0.0.0.0", port=8888, reload=True) diff --git a/test_app.py b/test_app.py index 37c7f7d..cee6b44 100644 --- a/test_app.py +++ b/test_app.py @@ -45,11 +45,11 @@ def test_pred_virginica2(): payload = { "sepal_length": 8, "sepal_width": 1, - "petal_length": 0.1, + "petal_length": 1, "petal_width": 8, } with TestClient(app) as client: response = client.post("/predict_flower", json=payload) # asserting the correct response is received assert response.status_code == 200 - assert response.json() == {"flower_class": "Iris Setosa"} \ No newline at end of file + assert response.json() == {"flower_class": "Iris Setosa"}