Update app.py

app.py

@@ -63,7 +63,7 @@ def update(inp, file, mode):
     print(voxels.shape)
     model = Model()
     model.to(device)
-    model.load_state_dict(torch.load("weights/metal_0.5A_v3_d0.2_16Abox.pth"))
+    model.load_state_dict(torch.load("weights/metal_0.5A_v3_d0.2_16Abox.pth", map_location=torch.device('cpu')))
     model.eval()
     with warnings.catch_warnings():
         warnings.filterwarnings("ignore")