Spaces:

stamps-labs
/

stamp2vec

Runtime error

sadjava commited on Jul 9, 2023

Commit

85b4b37

•

1 Parent(s): 05aeeac

Fixed for cpu

Files changed (1) hide show

app.py CHANGED Viewed

@@ -26,7 +26,7 @@ transform = A.Compose([
     ToTensorV2(p=1.0),
 ])
-vits8 = torch.jit.load(hf_hub_download('stamps-labs/vits8-stamp', filename='vits8stamp-torchscript.pth'))
 vits8 = vits8.to(device)
 vits8.eval()

     ToTensorV2(p=1.0),
 ])
+vits8 = torch.jit.load(hf_hub_download('stamps-labs/vits8-stamp', filename='vits8stamp-torchscript.pth'), map_location='cpu')
 vits8 = vits8.to(device)
 vits8.eval()