Keep 1 model (several leading to memory errors)

app.py

@@ -6,10 +6,11 @@ from transformers import CLIPProcessor, CLIPModel
 from st_clickable_images import clickable_images
 
 MODEL_NAMES = [
-    "base-patch32",
-    "base-patch16",
-    "large-patch14-336",
-]  # , "large-patch14", "large-patch14-336"]
+#    "base-patch32",
+#    "base-patch16",
+#    "large-patch14",    
+    "large-patch14-336"
+]
 
 
 @st.cache(allow_output_mutation=True)
@@ -157,9 +158,9 @@ def main():
     st.sidebar.markdown(description)
     with st.sidebar.expander("Advanced use"):
         st.markdown(howto)
-    mode = st.sidebar.selectbox(
-        "", ["Results for ViT-L/14@336px", "Comparison of 2 models"], index=0
-    )
+    #mode = st.sidebar.selectbox(
+    #    "", ["Results for ViT-L/14@336px", "Comparison of 2 models"], index=0
+    #)
 
     _, c, _ = st.columns((1, 3, 1))
     if "query" in st.session_state:
@@ -175,7 +176,7 @@ def main():
         "ViT-L/14@336px (slower)": "large-patch14-336",
     }
 
-    if "Comparison" in mode:
+    if False:#"Comparison" in mode:
         c1, c2 = st.columns((1, 1))
         selection1 = c1.selectbox("", models_dict.keys(), index=0)
         selection2 = c2.selectbox("", models_dict.keys(), index=2)
@@ -186,7 +187,7 @@ def main():
 
     if len(query) > 0:
         results1 = image_search(query, corpus, name1)
-        if "Comparison" in mode:
+        if False:#"Comparison" in mode:
             with c1:
                 clicked1 = clickable_images(
                     [result[0] for result in results1],
@@ -224,8 +225,8 @@ def main():
             if change_query:
                 if clicked1 >= 0:
                     st.session_state["query"] = f"[{corpus}:{results1[clicked1][2]}]"
-                elif clicked2 >= 0:
-                    st.session_state["query"] = f"[{corpus}:{results2[clicked2][2]}]"
+                #elif clicked2 >= 0:
+                #    st.session_state["query"] = f"[{corpus}:{results2[clicked2][2]}]"
                 st.experimental_rerun()