Update app.py

app.py

@@ -117,8 +117,8 @@ def retrieval_pc(load_data, k, sim_th, filter_fn):
     pc = load_data(prog)
     prog.progress(0.5, "Computing Embeddings")
     col2 = utils.render_pc(pc)
-    ref_dev = next(model_g14.parameters()).device
-    enc = model_g14(torch.tensor(pc[:, [0, 2, 1, 3, 4, 5]].T[None], device=ref_dev))
+    ref_dev = next(model_retrieval.parameters()).device
+    enc = model_retrieval(torch.tensor(pc[:, [0, 2, 1, 3, 4, 5]].T[None], device=ref_dev))
     if model_name == "pb-sn-M":
         enc = pc_adapter(enc)
     sim = torch.matmul(torch.nn.functional.normalize(lvis.feats, dim=-1), torch.nn.functional.normalize(enc.cpu(), dim=-1).squeeze())
@@ -237,6 +237,7 @@ try:
             if st.sidebar.button("submit"):
                 classification_custom(load_data, cats)
     elif task == "Cross-modal retrieval":
+        model_retrieval = load_tripletmix('tripletmix-pointbert-all-objaverse')
         input_mode = st.sidebar.selectbox(
             'Choose an input modality', 
             ("Point Cloud", "Image", "Text")