Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from gender_classification import gender_classification
 from emotion_classification import emotion_classification
+from dog_breed import dog_breed_classification
 
 # Functions to update the model state when a button is clicked.
 def select_gender():
@@ -9,27 +10,37 @@ def select_gender():
 def select_emotion():
     return "emotion"
 
+def select_dog_breed():
+    return "dog_breed"
+
 # Main classification function that calls the appropriate model based on selection.
 def classify(image, model_name):
     if model_name == "gender":
         return gender_classification(image)
     elif model_name == "emotion":
         return emotion_classification(image)
+    elif model_name == "dog_breed":
+        return dog_breed_classification(image)
     else:
         return {"Error": "No model selected"}
 
 with gr.Blocks() as demo:
     # Sidebar with title and model selection buttons.
     with gr.Sidebar():
-        gr.Markdown("# SigLIP2 Classification")
+        gr.Markdown("# SigLIP2 224")
         with gr.Row():
             gender_btn = gr.Button("Gender Classification")
             emotion_btn = gr.Button("Emotion Classification")
+            dog_breed_btn = gr.Button("Dog Breed Classification")
+        
         # State to hold the current model choice.
         selected_model = gr.State("gender")
+
         # Set model state when buttons are clicked.
         gender_btn.click(fn=select_gender, inputs=[], outputs=selected_model)
         emotion_btn.click(fn=select_emotion, inputs=[], outputs=selected_model)
+        dog_breed_btn.click(fn=select_dog_breed, inputs=[], outputs=selected_model)
+
         gr.Markdown("### Current Model:")
         model_display = gr.Textbox(value="gender", interactive=False)
         # Update display when state changes.