Update demo_support/retrieval.py

demo_support/retrieval.py

@@ -23,9 +23,14 @@ meta = json.load(
 #     "glb": "glbs/000-000/94db219c315742909fee67deeeacae15.glb"
 # }
 meta = {x['u']: x for x in meta['entries']}
+"""
 deser = torch.load(
     hf_hub_download("OpenShape/openshape-objaverse-embeddings", "objaverse.pt", repo_type='dataset'), map_location='cpu'
 )
+"""
+deser = torch.load(
+    hf_hub_download("TripletMix/tripletmix-objaverse-embeddings", "objaverse.pt", repo_type='dataset'), map_location='cpu'
+)
 us = deser['us']
 feats = deser['feats']