Spaces:
Runtime error
Runtime error
Fixed for cpu
Browse files
app.py
CHANGED
@@ -26,7 +26,7 @@ transform = A.Compose([
|
|
26 |
ToTensorV2(p=1.0),
|
27 |
])
|
28 |
|
29 |
-
vits8 = torch.jit.load(hf_hub_download('stamps-labs/vits8-stamp', filename='vits8stamp-torchscript.pth'))
|
30 |
vits8 = vits8.to(device)
|
31 |
vits8.eval()
|
32 |
|
|
|
26 |
ToTensorV2(p=1.0),
|
27 |
])
|
28 |
|
29 |
+
vits8 = torch.jit.load(hf_hub_download('stamps-labs/vits8-stamp', filename='vits8stamp-torchscript.pth'), map_location='cpu')
|
30 |
vits8 = vits8.to(device)
|
31 |
vits8.eval()
|
32 |
|