edit

main.py

@@ -2,7 +2,6 @@
 from fastapi import FastAPI
 from pydantic import BaseModel
 import joblib
-from typing import Dict
 
 app = FastAPI()
 loaded_model = joblib.load('model.joblib', mmap_mode='r')
@@ -31,17 +30,18 @@ class PredictResponse(BaseModel):
 
 
 
+
 @app.post('/predict', response_model=PredictResponse)
 def predict(data: PredictRequest):
     features = [data.AG_X, data.AG_Y, data.AG_Z, data.Acc_X, data.Acc_Y, data.Acc_Z, data.Gravity_X, data.Gravity_Y,
                 data.Gravity_Z, data.RR_X, data.RR_Y, data.RR_Z, data.RV_X, data.RV_Y, data.RV_Z, data.cos]
 
-    predictions = loaded_model.predict([features])[0]
+    predictions = loaded_model.predict([features])
 
 
     activities = ["Running", "Sitting", "Standing", "Walking", "DownStairs", "Upstairs"]
     activity = activities[predictions]
 
-    return {'activity': activity}
+    return {'activity': predictions}