Update app.py

app.py

@@ -4,7 +4,7 @@ import numpy as np
 import mediapipe as mp
 from sklearn.linear_model import LinearRegression
 import random
-import joblib
+import pickle
 
 # Setup for Face Mesh detection
 mp_face_mesh = mp.solutions.face_mesh
@@ -31,10 +31,15 @@ def train_model(output_range):
     model = LinearRegression().fit(X, y)
     return model
 
-# Load pre-trained models for Hemoglobin, SPO2, and Heart Rate
-hemoglobin_model = joblib.load("hemoglobin_model_from_anemia_dataset.pkl")
-spo2_model = joblib.load("spo2_model_simulated.pkl")
-hr_model = joblib.load("heart_rate_model.pkl")
+# Load pre-trained models for Hemoglobin, SPO2, and Heart Rate using pickle
+with open("hemoglobin_model_from_anemia_dataset.pkl", "rb") as f:
+    hemoglobin_model = pickle.load(f)
+
+with open("spo2_model_simulated.pkl", "rb") as f:
+    spo2_model = pickle.load(f)
+
+with open("heart_rate_model.pkl", "rb") as f:
+    hr_model = pickle.load(f)
 
 # Model dictionary setup for other tests
 models = {