Update app.py

app.py

@@ -12,6 +12,7 @@ from indian_western_food_classify import food_classification
 from bird_species import bird_classification
 from alphabet_sign_language_detection import sign_language_classification
 from rice_leaf_disease import classify_leaf_disease
+from traffic_density import traffic_density_classification
 
 # Main classification function that calls the appropriate model based on selection.
 def classify(image, model_name):
@@ -41,6 +42,8 @@ def classify(image, model_name):
         return classify_leaf_disease(image)
     elif model_name == "sign language":
         return sign_language_classification(image)
+    elif model_name == "traffic density":
+        return traffic_density_classification(image)
     else:
         return {"Error": "No model selected"}
 
@@ -50,7 +53,7 @@ def select_model(model_name):
         "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", "leaf disease": "secondary",
-        "sign language": "secondary"
+        "sign language": "secondary", "traffic density": "secondary"
     }
     model_variants[model_name] = "primary"
     return (model_name, *(gr.update(variant=model_variants[key]) for key in model_variants))
@@ -67,19 +70,20 @@ 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")
-            food_btn = gr.Button("Indian/Western Food", variant="secondary")
-            bird_btn = gr.Button("Bird Species", variant="secondary")
-            leaf_disease_btn = gr.Button("Rice Leaf Disease", variant="secondary")
-            sign_language_btn = gr.Button("Sign Language", variant="secondary")
+            fashion_mnist_btn = gr.Button("Fashion MNIST Classification", 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")
+            sign_language_btn = gr.Button("Sign Language Detection", variant="secondary")
+            traffic_density_btn = gr.Button("Traffic Density 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, leaf_disease_btn, sign_language_btn]
-        model_names = ["gender", "emotion", "dog breed", "deepfake", "gym workout", "waste", "age", "mnist", "fashion_mnist", "food", "bird", "leaf disease", "sign language"]
+        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, sign_language_btn, traffic_density_btn]
+        model_names = ["gender", "emotion", "dog breed", "deepfake", "gym workout", "waste", "age", "mnist", "fashion_mnist", "food", "bird", "leaf disease", "sign language", "traffic density"]
         
         for btn, name in zip(buttons, model_names):
             btn.click(fn=lambda n=name: select_model(n), inputs=[], outputs=[selected_model] + buttons)