Update app.py

app.py

@@ -8,6 +8,7 @@ from augmented_waste_classifier import waste_classification
 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
 
 # Main classification function that calls the appropriate model based on selection.
 def classify(image, model_name):
@@ -29,6 +30,8 @@ def classify(image, model_name):
         return classify_digit(image)
     elif model_name == "fashion_mnist":
         return fashion_mnist_classification(image)
+    elif model_name == "food":
+        return food_classification(image)
     else:
         return {"Error": "No model selected"}
 
@@ -36,7 +39,8 @@ def classify(image, model_name):
 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"
+        "gym workout": "secondary", "waste": "secondary", "age": "secondary", "mnist": "secondary",
+        "fashion_mnist": "secondary", "food": "secondary"
     }
     model_variants[model_name] = "primary"
     return (model_name, *(gr.update(variant=model_variants[key]) for key in model_variants))
@@ -53,15 +57,16 @@ with gr.Blocks() as demo:
             gym_workout_btn = gr.Button("Gym Workout Classification", variant="secondary")
             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", variant="secondary")
+            fashion_mnist_btn = gr.Button("Fashion MNIST Classification", variant="secondary")
+            food_btn = gr.Button("Indian/Western Food", 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]
-        model_names = ["gender", "emotion", "dog breed", "deepfake", "gym workout", "waste", "age", "mnist", "fashion_mnist"]
+        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"]
         
         for btn, name in zip(buttons, model_names):
             btn.click(fn=lambda n=name: select_model(n), inputs=[], outputs=[selected_model] + buttons)