Update app.py

Add weights_only parameter for latest pytorch 2.6

app.py

@@ -16,7 +16,7 @@ gc.enable()
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 
 with st.spinner("Loading PointNet++ model..."):
-    checkpoint = torch.load('checkpoints/best_model.pth', map_location=torch.device(device))
+    checkpoint = torch.load('checkpoints/best_model.pth', map_location=torch.device(device), weights_only=True)
     classifier = pn2.get_model(num_class=4, normal_channel=False)
     classifier.load_state_dict(checkpoint['model_state_dict'])
     classifier.eval()