Update app.py

app.py

@@ -46,7 +46,7 @@ sam_ffhq_aging = hf_hub_download(repo_id="AshanGimhana/Face_Agin_model", filenam
 new_age_model = load_model("age_prediction_model.h5")
 
 # 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()
@@ -112,11 +112,8 @@ def get_mouth_region(image):
         return np.array(mouth_points, np.int32)
     return None
 
+
 # Function to predict age
-def get_age(distr):
-    # Convert distribution to approximate age by scaling
-    age = distr * 4
-    return age
 
 def predict_age(image):
     image = np.array(image.resize((64, 64)))