Update app.py

app.py

@@ -22,7 +22,8 @@ from alphabet_sign_language_detection import sign_language_classification
 from rice_leaf_disease import classify_leaf_disease
 from traffic_density import traffic_density_classification
 from clip_art import clipart_classification
-from multisource_121 import multisource_classification  # New import
+from multisource_121 import multisource_classification
+from painting_126 import painting_classification  # New import
 
 # Gradio-Theme
 class Seafoam(Base):
@@ -97,6 +98,8 @@ def classify(image, model_name):
         return clipart_classification(image)
     elif model_name == "multisource":
         return multisource_classification(image)
+    elif model_name == "painting":  # New option
+        return painting_classification(image)
     else:
         return {"Error": "No model selected"}
 
@@ -107,7 +110,7 @@ def select_model(model_name):
         "gym workout": "secondary", "waste": "secondary", "age": "secondary", "mnist": "secondary",
         "fashion_mnist": "secondary", "food": "secondary", "bird": "secondary", "leaf disease": "secondary",
         "sign language": "secondary", "traffic density": "secondary", "clip art": "secondary",
-        "multisource": "secondary"  # New model variant
+        "multisource": "secondary", "painting": "secondary"  # New model variant
     }
     model_variants[model_name] = "primary"
     return (model_name, *(gr.update(variant=model_variants[key]) for key in model_variants))
@@ -175,7 +178,8 @@ with gr.Blocks(theme=seafoam) as demo:
                     sign_language_btn = gr.Button("Alphabet Sign Language", variant="secondary")
                     traffic_density_btn = gr.Button("Traffic Density", variant="secondary")
                     clip_art_btn = gr.Button("Art Classification", variant="secondary")
-                    multisource_btn = gr.Button("Multi-Source Classification", variant="secondary")  # New button
+                    multisource_btn = gr.Button("Multi-Source Classification", variant="secondary")
+                    painting_btn = gr.Button("Painting Classification", variant="secondary")  # New button
                 
                 selected_model = gr.State("age")
                 gr.Markdown("### Current Model:")
@@ -185,12 +189,12 @@ with gr.Blocks(theme=seafoam) as demo:
                 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, clip_art_btn, multisource_btn  # Include new button
+                    sign_language_btn, traffic_density_btn, clip_art_btn, multisource_btn, painting_btn  # Include new button
                 ]
                 model_names = [
                     "gender", "emotion", "dog breed", "deepfake", "gym workout", "waste",
                     "age", "mnist", "fashion_mnist", "food", "bird", "leaf disease",
-                    "sign language", "traffic density", "clip art", "multisource"  # New model name
+                    "sign language", "traffic density", "clip art", "multisource", "painting"  # New model name
                 ]
 
                 for btn, name in zip(buttons, model_names):
@@ -216,4 +220,4 @@ with gr.Blocks(theme=seafoam) as demo:
                     siglip2_output = gr.Label(label="SigLIP 2 Output", num_top_classes=3)
             zs_run_button.click(fn=infer, inputs=[zs_image_input, zs_text_input], outputs=[siglip1_output, siglip2_output])
 
-demo.launch()
+demo.launch()