Update app.py

app.py

@@ -32,6 +32,14 @@ def load_openshape(name, to_cpu=False):
     return pce
 
 
+def load_tripletmix(name, to_cpu=False):
+    pce = openshape.load_pc_encoder_mix(name)
+    if to_cpu:
+        pce = pce.cpu()
+    return pce
+
+
+
 def retrieval_filter_expand():
     sim_th = st.sidebar.slider("Similarity Threshold", 0.05, 0.5, 0.1, key='rsimth')
     tag = ""
@@ -189,7 +197,8 @@ try:
     f32 = numpy.float32
     half = torch.float16 if torch.cuda.is_available() else torch.bfloat16
     clip_model, clip_prep = load_openclip()
-    model_g14 = load_openshape('openshape-pointbert-vitg14-rgb')
+    #model_g14 = load_openshape('openshape-pointbert-vitg14-rgb')
+    model_g14 = load_tripletmix('tripletmix-spconv-all')
 
     st.caption("This demo presents three tasks: 3D classification, cross-modal retrieval, and cross-modal generation. Examples are provided for demonstration purposes. You're encouraged to fine-tune task parameters and upload files for customized testing as required.")
     st.sidebar.title("TripletMix Demo Configuration Panel")