Update app.py

app.py

@@ -52,18 +52,25 @@ age_calc_model = torch.load('Custom_Age_prediction_model.pth')
 ########################################################################
 
 # 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")
-sam_ffhq_aging = hf_hub_download(repo_id="AshanGimhana/Face_Agin_model", filename="sam_ffhq_aging.pt")
+#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")
+
 
 ########################################################################
 ############## caffe model for age calculation #######################
 
+#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 prediction model setup
 #age_net = cv2.dnn.readNetFromCaffe(age_prototxt, caffe_model)
 ########################################################################
 
 
+########################################################################################################
+# Aging model #
+sam_ffhq_aging = hf_hub_download(repo_id="AshanGimhana/Face_Agin_model", filename="sam_ffhq_aging.pt")
+########################################################################################################
+
 # Face detection and landmarks predictor setup
 detector = dlib.get_frontal_face_detector()
 predictor = dlib.shape_predictor("shape_predictor_68_face_landmarks.dat")