Update app.py

app.py

@@ -6,6 +6,7 @@ import numpy as np
 import mediapipe as mp
 from sklearn.linear_model import LinearRegression
 import random
+import joblib
 
 mp_face_mesh = mp.solutions.face_mesh
 face_mesh = mp_face_mesh.FaceMesh(static_image_mode=True, max_num_faces=1, refine_landmarks=True, min_detection_confidence=0.5)
@@ -29,13 +30,13 @@ def train_model(output_range):
     model = LinearRegression().fit(X, y)
     return model
 
-import joblib
+# Load real models
 hemoglobin_model = joblib.load("hemoglobin_model_from_anemia_dataset.pkl")
-
-hemoglobin_r2 = 0.385
-import joblib
 spo2_model = joblib.load("spo2_model_simulated.pkl")
 hr_model = joblib.load("heart_rate_model.pkl")
+tsh_model = joblib.load("tsh_model.pkl")
+
+hemoglobin_r2 = 0.385
 
 models = {
     "Hemoglobin": hemoglobin_model,
@@ -49,7 +50,7 @@ models = {
     "Urea": train_model((7, 20)),
     "Sodium": train_model((135, 145)),
     "Potassium": train_model((3.5, 5.1)),
-    "TSH": train_model((0.4, 4.0)),
+    "TSH": tsh_model,
     "Cortisol": train_model((5, 25)),
     "FBS": train_model((70, 110)),
     "HbA1c": train_model((4.0, 5.7)),