Update app.py

app.py

@@ -10,6 +10,7 @@ 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
+from rice_leaf_disease import classify_leaf_disease
 
 # Main classification function that calls the appropriate model based on selection.
 def classify(image, model_name):
@@ -35,6 +36,8 @@ def classify(image, model_name):
         return food_classification(image)
     elif model_name == "bird":
         return bird_classification(image)
+    elif model_name == "leaf disease":
+        return classify_leaf_disease(image)
     else:
         return {"Error": "No model selected"}
 
@@ -43,7 +46,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", "bird": "secondary"
+        "fashion_mnist": "secondary", "food": "secondary", "bird": "secondary", "leaf disease": "secondary"
     }
     model_variants[model_name] = "primary"
     return (model_name, *(gr.update(variant=model_variants[key]) for key in model_variants))
@@ -61,16 +64,17 @@ with gr.Blocks() as demo:
             waste_btn = gr.Button("Waste Classification", variant="secondary")
             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")
+            food_btn = gr.Button("Indian/Western Food Classification", variant="secondary")
+            bird_btn = gr.Button("Bird Species Classification", variant="secondary")
+            leaf_disease_btn = gr.Button("Rice Leaf Disease Classification", 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, bird_btn]
-        model_names = ["gender", "emotion", "dog breed", "deepfake", "gym workout", "waste", "age", "mnist", "fashion_mnist", "food", "bird"]
+        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, leaf_disease_btn]
+        model_names = ["gender", "emotion", "dog breed", "deepfake", "gym workout", "waste", "age", "mnist", "fashion_mnist", "food", "bird", "leaf disease"]
         
         for btn, name in zip(buttons, model_names):
             btn.click(fn=lambda n=name: select_model(n), inputs=[], outputs=[selected_model] + buttons)