Spaces:

matikosowy
/

Photo_Colorizer

Runtime error

matikosowy commited on Sep 10, 2024

Commit

f618c7c

verified ·

1 Parent(s): 2add093

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -84,7 +84,7 @@ class DummyModel(nn.Module):
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 model = DummyModel()
-model.load_state_dict(torch.load('model.pth', weights_only=True))
 model = model.to(device)
 model.eval()

 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 model = DummyModel()
+model.load_state_dict(torch.load('model.pth'))
 model = model.to(device)
 model.eval()