Update app.py

app.py

@@ -23,7 +23,8 @@ 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
-from painting_126 import painting_classification  # New import
+from painting_126 import painting_classification
+from sketch_126 import sketch_classification  # New import
 
 # Gradio-Theme
 class Seafoam(Base):
@@ -32,7 +33,7 @@ class Seafoam(Base):
         *,
         primary_hue: colors.Color | str = colors.emerald,
         secondary_hue: colors.Color | str = colors.blue,
-        neutral_hue: colors.Color | str = colors.gray,
+        neutral_hue: colors.Color | str = colors.blue,
         spacing_size: sizes.Size | str = sizes.spacing_md,
         radius_size: sizes.Size | str = sizes.radius_md,
         text_size: sizes.Size | str = sizes.text_lg,
@@ -61,6 +62,21 @@ class Seafoam(Base):
             font=font,
             font_mono=font_mono,
         )
+        super().set(
+            body_background_fill="repeating-linear-gradient(45deg, *primary_200, *primary_200 10px, *primary_50 10px, *primary_50 20px)",
+            body_background_fill_dark="repeating-linear-gradient(45deg, *primary_800, *primary_800 10px, *primary_900 10px, *primary_900 20px)",
+            button_primary_background_fill="linear-gradient(90deg, *primary_300, *secondary_400)",
+            button_primary_background_fill_hover="linear-gradient(90deg, *primary_200, *secondary_300)",
+            button_primary_text_color="white",
+            button_primary_background_fill_dark="linear-gradient(90deg, *primary_600, *secondary_800)",
+            slider_color="*secondary_300",
+            slider_color_dark="*secondary_600",
+            block_title_text_weight="600",
+            block_border_width="3px",
+            block_shadow="*shadow_drop_lg",
+            button_primary_shadow="*shadow_drop_lg",
+            button_large_padding="32px",
+        )
 
 seafoam = Seafoam()
 
@@ -98,8 +114,10 @@ def classify(image, model_name):
         return clipart_classification(image)
     elif model_name == "multisource":
         return multisource_classification(image)
-    elif model_name == "painting":  # New option
+    elif model_name == "painting":
         return painting_classification(image)
+    elif model_name == "sketch":  # New option
+        return sketch_classification(image)
     else:
         return {"Error": "No model selected"}
 
@@ -110,13 +128,12 @@ 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", "painting": "secondary"  # New model variant
+        "multisource": "secondary", "painting": "secondary", "sketch": "secondary"  # New model variant
     }
     model_variants[model_name] = "primary"
     return (model_name, *(gr.update(variant=model_variants[key]) for key in model_variants))
 
 # Zero-Shot Classification Setup (SigLIP models)
-# Load the SigLIP models and processors
 sg1_ckpt = "google/siglip-so400m-patch14-384"
 siglip1_model = AutoModel.from_pretrained(sg1_ckpt, device_map="cpu").eval()
 siglip1_processor = AutoProcessor.from_pretrained(sg1_ckpt)
@@ -125,7 +142,6 @@ sg2_ckpt = "google/siglip2-so400m-patch14-384"
 siglip2_model = AutoModel.from_pretrained(sg2_ckpt, device_map="cpu").eval()
 siglip2_processor = AutoProcessor.from_pretrained(sg2_ckpt)
 
-# Utilities for zero-shot classification.
 @spaces.GPU
 def postprocess_siglip(sg1_probs, sg2_probs, labels):
     sg1_output = {labels[i]: sg1_probs[0][i].item() for i in range(len(labels))}
@@ -166,21 +182,22 @@ with gr.Blocks(theme=seafoam) as demo:
                     age_btn = gr.Button("Age Classification", variant="primary")
                     gender_btn = gr.Button("Gender Classification", variant="secondary")
                     emotion_btn = gr.Button("Emotion Classification", variant="secondary")
-                    dog_breed_btn = gr.Button("Dog Breed Classification", variant="secondary")
-                    deepfake_btn = gr.Button("Deepfake vs Real", variant="secondary")
                     gym_workout_btn = gr.Button("Gym Workout Classification", variant="secondary")
+                    dog_breed_btn = gr.Button("Dog Breed Classification", variant="secondary")
+                    bird_btn = gr.Button("Bird Species Classification", variant="secondary")
                     waste_btn = gr.Button("Waste Classification", variant="secondary")
+                    traffic_density_btn = gr.Button("Traffic Density", variant="secondary")
+                    sign_language_btn = gr.Button("Alphabet Sign Language", variant="secondary")
+                    clip_art_btn = gr.Button("Clip Art 126", 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 Classification", variant="secondary")
+                    fashion_mnist_btn = gr.Button("Fashion MNIST (only cloth)", variant="secondary")
+                    food_btn = gr.Button("Indian/Western Food Type", variant="secondary")
                     leaf_disease_btn = gr.Button("Rice Leaf Disease", variant="secondary")
-                    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")
-                    painting_btn = gr.Button("Painting Classification", variant="secondary")  # New button
-                
+                    multisource_btn = gr.Button("Multi Source 121", variant="secondary")
+                    painting_btn = gr.Button("Painting 126", variant="secondary")
+                    sketch_btn = gr.Button("Sketch 126", variant="secondary")
+                    deepfake_btn = gr.Button("Deepfake vs Real", variant="secondary")
+                    
                 selected_model = gr.State("age")
                 gr.Markdown("### Current Model:")
                 model_display = gr.Textbox(value="age", interactive=False)
@@ -189,12 +206,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, painting_btn  # Include new button
+                    sign_language_btn, traffic_density_btn, clip_art_btn, multisource_btn, painting_btn, sketch_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", "painting"  # New model name
+                    "sign language", "traffic density", "clip art", "multisource", "painting", "sketch"  # New model name
                 ]
 
                 for btn, name in zip(buttons, model_names):