Update app.py

app.py

@@ -235,7 +235,15 @@ try:
             if st.sidebar.button("submit"):
                 classification_custom(load_data, cats)
     elif task == "Cross-modal retrieval":
-        model_retrieval = load_tripletmix('tripletmix-pointbert-all-objaverse')
+        #model_retrieval = load_tripletmix('tripletmix-pointbert-all-objaverse')
+        model_name = st.sidebar.selectbox(
+            'Model Selection',
+            ("pb-Mix", "pb")
+        )
+        if model_name == "pb-Mix":
+            model_retrieval = load_tripletmix('tripletmix-pointbert-all-objaverse')
+        elif model_name == "pb":
+            model_retrieval = load_openshape('openshape-pointbert-vitg14-rgb')
         input_mode = st.sidebar.selectbox(
             'Choose an input modality', 
             ("Point Cloud", "Image", "Text")