test3

app.py

@@ -30,14 +30,14 @@ async def predict_and_add(data: dict):
         raise HTTPException(status_code=400, detail="Please provide correct values")
 
     # Scale and predict
-    input_bagemann = np.array([[HL, HB]])
+    input_bagemann = np.array([[HB,HL]])
     bagemann_scaled = scaler_bagemann.transform(input_bagemann)
     prediction_bagemann = model_bagemann.predict(bagemann_scaled)
     predicted_class_bagemann = int(np.argmax(prediction_bagemann, axis=1)[0])
     
        # Scale and predict
     input_shertmann = np.array([[HB,FR]])
-    shertmann_scaled = scaler_bagemann.transform(input_shertmann)
+    shertmann_scaled = scaler_shertmann.transform(input_shertmann)
     prediction_shertmann = model_shertmann.predict(shertmann_scaled)
     predicted_class_shertmann = int(np.argmax(prediction_shertmann, axis=1)[0])