Update main.py

main.py

@@ -69,14 +69,21 @@ 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):
+#@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}
+   # 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}
 
 if __name__ == "__main__":
     os.environ["DEBUG"] = "True"  # Enable debug mode