Update app.py

app.py

@@ -36,14 +36,15 @@ from models.psp import pSp
 # Huggingface login
 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")
+
 # 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")
 
-# If 'mse' is a custom function needed, 
-#custom_objects = {'mse': MeanSquaredError()}
-new_age_model = load_model("age_prediction_model.h5")
 
 # Age prediction model setup
 # age_net = cv2.dnn.readNetFromCaffe(age_prototxt, caffe_model)