Update app.py

app.py

@@ -8,11 +8,19 @@ import dlib
 import cv2
 from keras.preprocessing.image import img_to_array
 from keras.models import load_model
+from huggingface_hub import hf_hub_download
 
 # Define global variables
 points = []
 emotion_classifier = None
 
+# Define Hugging Face repository details
+repo_id = "jaimin/Age_detection"
+
+# Download model files from Hugging Face
+predictor_path = hf_hub_download(repo_id=repo_id, filename="shape_predictor_68_face_landmarks.dat")
+emotion_model_path = hf_hub_download(repo_id=repo_id, filename="XCEPTION.102-0.66.hdf5")
+
 def eye_brow_distance(leye, reye):
     global points
     distq = dist.euclidean(leye, reye)
@@ -41,9 +49,8 @@ def normalize_values(points, disp):
 def stress(video_path, duration):
     global points, emotion_classifier
     detector = dlib.get_frontal_face_detector()
-    predictor_path = "Stress/shape_predictor_68_face_landmarks.dat"
     predictor = dlib.shape_predictor(predictor_path)
-    emotion_classifier = load_model("Stress/XCEPTION.102-0.66.hdf5", compile=False)
+    emotion_classifier = load_model(emotion_model_path, compile=False)
     
     # Open video file
     cap = cv2.VideoCapture(video_path)
@@ -118,4 +125,3 @@ gr.Interface(
     outputs="json",
     title="Heart Rate and Stress Measurement"
 ).launch(server_name="0.0.0.0")
-