Test with only 2 small models

app.py

@@ -5,7 +5,10 @@ import pandas as pd, numpy as np
 from transformers import CLIPProcessor, CLIPModel
 from st_clickable_images import clickable_images
 
-MODEL_NAMES = ["base-patch32", "base-patch16", "large-patch14", "large-patch14-336"]
+MODEL_NAMES = [
+    "base-patch32",
+    "base-patch16",
+]  # , "large-patch14", "large-patch14-336"]
 
 
 @st.cache(show_spinner=False, hash_funcs={dict: lambda _: None})
@@ -174,7 +177,7 @@ def main():
     if "Comparison" in mode:
         c1, c2 = st.columns((1, 1))
         selection1 = c1.selectbox("", models_dict.keys(), index=0)
-        selection2 = c2.selectbox("", models_dict.keys(), index=3)
+        selection2 = c2.selectbox("", models_dict.keys(), index=1)
         name1 = models_dict[selection1]
         name2 = models_dict[selection2]
     else: