Update app.py

app.py

@@ -11,6 +11,9 @@ pca = joblib.load("pca.joblib")
 with open("cluster_fatigue_map.json") as f:
     cluster_fatigue_map = json.load(f)
 
+# Load trained regressor
+model = joblib.load("gb_regressor.joblib")
+
 feature_cols = [
     "AVRR", "SDNN", "RMSSD", "PNN50", "Coefficient_of_Variation",
     "Age", "Weight", "Height"
@@ -45,7 +48,7 @@ def debug_predict(input_data: FatigueInput):
         "scaler_scale": scaler.scale_.tolist()
     }
 
-@app.post("/predict")
+@app.post("/predictGMM")
 def predict(input_data: FatigueInput):
     try:
         input_dict = input_data.dict()
@@ -73,3 +76,18 @@ def predict(input_data: FatigueInput):
         }
     except Exception as e:
         return {"error": str(e)}
+
+@app.post("/predict")
+def predict(input_data: FatigueInput):
+    try:
+        input_df = pd.DataFrame([input_data.dict()])[feature_cols]
+
+        # Predict fatigue level
+        predicted_fatigue = model.predict(input_df)[0]
+
+        return {
+            "predicted_fatigue_level": float(predicted_fatigue),
+            "rounded_level": int(round(predicted_fatigue))
+        }
+    except Exception as e:
+        return {"error": str(e)}