Update app.py

app.py

@@ -52,8 +52,7 @@ try:
     spo2_model = joblib.load("spo2_model_simulated.pkl")
     hr_model = joblib.load("heart_rate_model.pkl")
 except FileNotFoundError:
-    print(
-        "Error: One or more .pkl model files are missing. Please upload them.")
+    print("Error: One or more .pkl model files are missing. Please upload them.")
     exit(1)
 
 models = {
@@ -241,8 +240,8 @@ def analyze_face(input_data):
         0].landmark  # Fixed: Use integer index
     features = extract_features(frame_rgb, landmarks)
     
-    # Convert features to pandas DataFrame if the model was trained with column names
-    features_df = pd.DataFrame([features], columns=["feature1", "feature2", "feature3"])
+    # Convert features to pandas DataFrame with correct column names
+    features_df = pd.DataFrame([features], columns=["%Red Pixel", "%Green Pixel", "%Blue Pixel"])
     
     test_values = {}
     r2_scores = {}