Update app.py

app.py

@@ -7,6 +7,7 @@ from gym_workout_classification import workout_classification
 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
 
 # Main classification function that calls the appropriate model based on selection.
 def classify(image, model_name):
@@ -26,6 +27,8 @@ def classify(image, model_name):
         return age_classification(image)
     elif model_name == "mnist":
         return classify_digit(image)
+    elif model_name == "fashion_mnist":
+        return fashion_mnist_classification(image)
     else:
         return {"Error": "No model selected"}
 
@@ -33,7 +36,7 @@ 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"
+        "gym workout": "secondary", "waste": "secondary", "age": "secondary", "mnist": "secondary", "fashion_mnist": "secondary"
     }
     model_variants[model_name] = "primary"
     return (model_name, *(gr.update(variant=model_variants[key]) for key in model_variants))
@@ -50,14 +53,15 @@ 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")
         
         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]
-        model_names = ["gender", "emotion", "dog breed", "deepfake", "gym workout", "waste", "age", "mnist"]
+        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"]
         
         for btn, name in zip(buttons, model_names):
             btn.click(fn=lambda n=name: select_model(n), inputs=[], outputs=[selected_model] + buttons)