Update app.py

app.py

@@ -1,10 +1,6 @@
 import gradio as gr
 import torch
-import time
 import spaces
-from typing import Iterable
-from gradio.themes.base import Base
-from gradio.themes.utils import colors, fonts, sizes
 from transformers import AutoModel, AutoProcessor
 
 from gender_classification import gender_classification
@@ -26,45 +22,6 @@ from multisource_121 import multisource_classification
 from painting_126 import painting_classification
 from sketch_126 import sketch_classification  # New import
 
-#Gradio-Theme
-class Seafoam(Base):
-    def __init__(
-        self,
-        *,
-        primary_hue: colors.Color | str = colors.emerald,
-        secondary_hue: colors.Color | str = colors.blue,
-        neutral_hue: colors.Color | str = colors.gray,
-        spacing_size: sizes.Size | str = sizes.spacing_md,
-        radius_size: sizes.Size | str = sizes.radius_md,
-        text_size: sizes.Size | str = sizes.text_lg,
-        font: fonts.Font
-        | str
-        | Iterable[fonts.Font | str] = (
-            fonts.GoogleFont("Quicksand"),
-            "ui-sans-serif",
-            "sans-serif",
-        ),
-        font_mono: fonts.Font
-        | str
-        | Iterable[fonts.Font | str] = (
-            fonts.GoogleFont("IBM Plex Mono"),
-            "ui-monospace",
-            "monospace",
-        ),
-    ):
-        super().__init__(
-            primary_hue=primary_hue,
-            secondary_hue=secondary_hue,
-            neutral_hue=neutral_hue,
-            spacing_size=spacing_size,
-            radius_size=radius_size,
-            text_size=text_size,
-            font=font,
-            font_mono=font_mono,
-        )
-
-seafoam = Seafoam()
-
 # Main classification function for multi-model classification.
 def classify(image, model_name):
     if model_name == "gender":
@@ -155,7 +112,7 @@ def infer(image, candidate_labels):
     return postprocess_siglip(sg1_probs, sg2_probs, labels=candidate_labels)
 
 # Build the Gradio Interface with two tabs.
-with gr.Blocks(theme=seafoam) as demo:
+with gr.Blocks(theme="allenai/gradio-theme") as demo:
     gr.Markdown("# Multi-Domain & Zero-Shot Image Classification")
     
     with gr.Tabs():