ML Update

ml/inference.py

@@ -47,7 +47,7 @@ diet_plan_mapping = {
 
 ##################################### LOAD MODELS #########################################
 
-risk_model = tf.keras.models.load_model('weights/risk_prediction.h5')
+risk_model = tf.keras.models.load_model('weights/future_prediction.h5')
 risk_model.compile(
     optimizer='adam',
     loss='categorical_crossentropy',
@@ -58,7 +58,7 @@ risk_model.compile(
         tf.keras.metrics.AUC()
     ])
 
-with open('weights/scalar - risk_prediction.pkl', 'rb') as f:
+with open('weights/scalar_future_prediction.pkl', 'rb') as f:
     risk_scalar = pickle.load(f)
 
 
@@ -70,7 +70,7 @@ diet_workout_model.compile(
     metrics=['accuracy']
 )
 
-with open('weights/scalar - diet_workout_prediction.pkl', 'rb') as f:
+with open('weights/scalar_diet_workout_prediction.pkl', 'rb') as f:
     diet_workout_scalar = pickle.load(f)
 
 ##################################### PREDICTION #########################################

[Serasinghe K.A.E.Y.D]

done