Update main.py

main.py

@@ -69,21 +69,14 @@ def root():
     return {"API": "This is an API for sepsis prediction."}
 
 # Prediction endpoint (Where we will input our features)
-#@app.post("/predict")
-#def predict_sepsis(patient: Patient):
-
-    # Make prediction
-   # data = pd.DataFrame(patient.dict(), index=[0])
-    #scaled_data = scaler.transform(data)
-    #parsed = predict(df=scaled_data)
-    #return {"output": parsed}
-
-
 @app.post("/predict")
 def predict_sepsis(patient: Patient):
-    patient_data = patient.dict()
-    prediction = predict(patient_data)
-    return {"output": prediction}
+
+    # Make prediction
+    data = pd.DataFrame(patient.dict(), index=[0])
+    scaled_data = scaler.transform(data)
+    parsed = predict(df=scaled_data)
+    return {"output": parsed}
 
 if __name__ == "__main__":
     os.environ["DEBUG"] = "True"  # Enable debug mode