Change model load to CPU

app.py

@@ -13,7 +13,7 @@ from SingleTreePointCloudLoader import SingleTreePointCloudLoader
 gc.enable()
 
 with st.spinner("Loading PointNet++ model..."):
-    checkpoint = torch.load('checkpoints/best_model.pth')
+    checkpoint = torch.load('checkpoints/best_model.pth', map_location=torch.device('cpu'))
     classifier = pn2.get_model(num_class=4, normal_channel=False)
     classifier.load_state_dict(checkpoint['model_state_dict'])
     classifier.eval()