Update app.py

app.py

@@ -38,16 +38,16 @@ login(token=os.getenv("TOKENKEY"))
 
 # If 'mse' is a custom function needed, 
 #custom_objects = {'mse': MeanSquaredError()}
-new_age_model = load_model("age_prediction_model.h5")
+#new_age_model = load_model("age_prediction_model.h5")
 
 # Download models from Huggingface
-#age_prototxt = hf_hub_download(repo_id="AshanGimhana/Age_Detection_caffe", filename="age.prototxt")
-#caffe_model = hf_hub_download(repo_id="AshanGimhana/Age_Detection_caffe", filename="dex_imdb_wiki.caffemodel")
+age_prototxt = hf_hub_download(repo_id="AshanGimhana/Age_Detection_caffe", filename="age.prototxt")
+caffe_model = hf_hub_download(repo_id="AshanGimhana/Age_Detection_caffe", filename="dex_imdb_wiki.caffemodel")
 sam_ffhq_aging = hf_hub_download(repo_id="AshanGimhana/Face_Agin_model", filename="sam_ffhq_aging.pt")
 
 
 # Age prediction model setup
-# age_net = cv2.dnn.readNetFromCaffe(age_prototxt, caffe_model)
+age_net = cv2.dnn.readNetFromCaffe(age_prototxt, caffe_model)
 
 # Face detection and landmarks predictor setup
 detector = dlib.get_frontal_face_detector()