Update app.py

app.py

@@ -5,6 +5,7 @@ from dog_breed import dog_breed_classification
 from deepfake_vs_real import deepfake_classification
 from gym_workout_classification import workout_classification
 from augmented_waste_classifier import waste_classification
+from age_classification import age_classification
 
 # Functions to update the model state when a button is clicked.
 def select_gender():
@@ -25,6 +26,9 @@ def select_gym_workout():
 def select_waste():
     return "waste"
 
+def select_age():
+    return "age"
+
 # Main classification function that calls the appropriate model based on selection.
 def classify(image, model_name):
     if model_name == "gender":
@@ -36,9 +40,11 @@ def classify(image, model_name):
     elif model_name == "deepfake":
         return deepfake_classification(image)
     elif model_name == "gym workout":
-        return gym_workout_classification(image)
+        return workout_classification(image)
     elif model_name == "waste":
         return waste_classification(image)
+    elif model_name == "age":
+        return age_classification(image)
     else:
         return {"Error": "No model selected"}
 
@@ -47,6 +53,7 @@ with gr.Blocks() as demo:
     with gr.Sidebar():
         gr.Markdown("# SigLIP2 224")
         with gr.Row():
+            age_btn = gr.Button("Age Classification")
             gender_btn = gr.Button("Gender Classification")
             emotion_btn = gr.Button("Emotion Classification")
             dog_breed_btn = gr.Button("Dog Breed Classification")
@@ -55,7 +62,7 @@ with gr.Blocks() as demo:
             waste_btn = gr.Button("Waste Classification")
 
         # State to hold the current model choice.
-        selected_model = gr.State("gender")
+        selected_model = gr.State("age")
 
         # Set model state when buttons are clicked.
         gender_btn.click(fn=select_gender, inputs=[], outputs=selected_model)
@@ -64,6 +71,7 @@ with gr.Blocks() as demo:
         deepfake_btn.click(fn=select_deepfake, inputs=[], outputs=selected_model)
         gym_workout_btn.click(fn=select_gym_workout, inputs=[], outputs=selected_model)
         waste_btn.click(fn=select_waste, inputs=[], outputs=selected_model)
+        age_btn.click(fn=select_age, inputs=[], outputs=selected_model)
 
         gr.Markdown("### Current Model:")
         model_display = gr.Textbox(value="gender", interactive=False)