Spaces:

sadjava
/

dog-localization

Runtime error

sadjava commited on Jun 7, 2023

Commit

ef44018

•

1 Parent(s): caa33e3

fixed for cpu

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ import cv2
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model = Model()
-model.load_state_dict(torch.load('model.pt'))
 model = model.to(device)
 model.eval()

 device = "cuda" if torch.cuda.is_available() else "cpu"
 model = Model()
+model.load_state_dict(torch.load('model.pt', map_location=torch.device('cpu')))
 model = model.to(device)
 model.eval()