Add the larger models

app.py

@@ -5,10 +5,7 @@ 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(allow_output_mutation=True)
@@ -177,7 +174,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=1)
+        selection2 = c2.selectbox("", models_dict.keys(), index=3)
         name1 = models_dict[selection1]
         name2 = models_dict[selection2]
     else: