Spaces:

simonduerr
/

metal3d

Runtime error

simonduerr commited on Jun 2, 2022

Commit

91fa11f

1 Parent(s): 61855d2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -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.eval()
     with warnings.catch_warnings():
         warnings.filterwarnings("ignore")

     print(voxels.shape)
     model = Model()
     model.to(device)
+    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")