Update app.py

app.py

@@ -9,6 +9,7 @@ from age_classification import age_classification
 from mnist_digits import classify_digit
 from fashion_mnist_cloth import fashion_mnist_classification
 from indian_western_food_classify import food_classification
+from bird_species import bird_classification
 
 # Main classification function that calls the appropriate model based on selection.
 def classify(image, model_name):
@@ -32,6 +33,8 @@ def classify(image, model_name):
         return fashion_mnist_classification(image)
     elif model_name == "food":
         return food_classification(image)
+    elif model_name == "bird":
+        return bird_classification(image)
     else:
         return {"Error": "No model selected"}
 
@@ -40,7 +43,7 @@ def select_model(model_name):
     model_variants = {
         "gender": "secondary", "emotion": "secondary", "dog breed": "secondary", "deepfake": "secondary",
         "gym workout": "secondary", "waste": "secondary", "age": "secondary", "mnist": "secondary",
-        "fashion_mnist": "secondary", "food": "secondary"
+        "fashion_mnist": "secondary", "food": "secondary", "bird": "secondary"
     }
     model_variants[model_name] = "primary"
     return (model_name, *(gr.update(variant=model_variants[key]) for key in model_variants))
@@ -59,14 +62,15 @@ with gr.Blocks() as demo:
             mnist_btn = gr.Button("Digit Classify (0-9)", variant="secondary")
             fashion_mnist_btn = gr.Button("Fashion MNIST Classification", variant="secondary")
             food_btn = gr.Button("Indian/Western Food", variant="secondary")
+            bird_btn = gr.Button("Bird Species", variant="secondary")
         
         selected_model = gr.State("age")
         gr.Markdown("### Current Model:")
         model_display = gr.Textbox(value="age", interactive=False)
         selected_model.change(lambda m: m, selected_model, model_display)
 
-        buttons = [gender_btn, emotion_btn, dog_breed_btn, deepfake_btn, gym_workout_btn, waste_btn, age_btn, mnist_btn, fashion_mnist_btn, food_btn]
-        model_names = ["gender", "emotion", "dog breed", "deepfake", "gym workout", "waste", "age", "mnist", "fashion_mnist", "food"]
+        buttons = [gender_btn, emotion_btn, dog_breed_btn, deepfake_btn, gym_workout_btn, waste_btn, age_btn, mnist_btn, fashion_mnist_btn, food_btn, bird_btn]
+        model_names = ["gender", "emotion", "dog breed", "deepfake", "gym workout", "waste", "age", "mnist", "fashion_mnist", "food", "bird"]
         
         for btn, name in zip(buttons, model_names):
             btn.click(fn=lambda n=name: select_model(n), inputs=[], outputs=[selected_model] + buttons)