done

main.py

@@ -2,6 +2,8 @@ from fastapi import FastAPI
 import pickle
 import uvicorn
 import pandas as pd
+import json
+import time
 
 app = FastAPI(debug=True)
 
@@ -19,14 +21,16 @@ ml_model = load_pickle('my_model.pkl')
 #Root endpoints
 @app.get("/")
 def root():
-    return {"API": "An API for Sepsis Prediction."}
+    return {"API": "An API for AMES Prediction."}
     
-@app.get('/Predict_Sepsis')
+@app.post('/predict')
 async def predict(data: dict):
     
-    data = pd.DataFrame(data)
-    print('data', data)
-    
-    model_output = ml_model.predict(data)
-    print('pred', model_output)
-    return model_output
+    start = time.time()
+    data_ = pd.DataFrame(data)
+    print('Data: ', type(data_))
+    model_output = ml_model.predict(data_)
+    print('Predictions: ', 10 ** model_output)
+    end = time.time()
+    time_to_inference = end - start
+    return {'predict': list(10 ** model_output), 'time_to_inference': time_to_inference}