Update app.py

app.py

@@ -35,8 +35,11 @@ MODEL_CACHE = {}
 MODEL_OPTIONS = {
     "CLIP-base":    "openai/clip-vit-base-patch32",
     "CLIP-large":   "openai/clip-vit-large-patch14",
-    "SigLIP-base":  "google/siglip-base-patch16-224",
-    "SigLIP-large": "google/siglip-so400m-patch14-384"}
+    # "SigLIP-base":  "google/siglip-base-patch16-224",
+    # "SigLIP-large": "google/siglip-so400m-patch14-384"
+    "ALIGN": "kakaobrain/align-base",
+    "LLM2CLIP": "microsoft/LLM2CLIP-EVA02-L-14-336"
+}
 
 CANDIDATE_LABELS = ["a bat (baseball)", "a bat (mammal)",
                     "a flower (plant)", "flour (baking powder)",
@@ -109,7 +112,7 @@ example_list = [
 
 gradio_app = gradio.Interface(
     fn     = run_classifer,
-    inputs = [gradio.Dropdown(["CLIP-base", "CLIP-large", "SigLIP-base", "SigLIP-large"], value="CLIP-large", label  = "Select Classifier"),
+    inputs = [gradio.Dropdown(["CLIP-base", "CLIP-large", "ALIGN", "LLM2CLIP"], value="CLIP-large", label  = "Select Classifier"),
               gradio.Image(type="pil", label="Load sample image here"),
               gradio.Slider(minimum = 0.1, maximum = 0.9, step = 0.05, value = 0.25, label = "Set Prediction Threshold")
               ],