fadindashfr commited on
Commit
b77752f
·
1 Parent(s): 17a1b09

fix RuntimeError

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ parser.read_meta(f="configs/metadata.json") # read the metadata from specified J
19
  inference = parser.get_parsed_content("inferer")
20
  network = parser.get_parsed_content("network_def")
21
  preprocess = parser.get_parsed_content("preprocessing")
22
- state_dict = torch.load("models/model.pt")
23
  network.load_state_dict(state_dict, strict=True) # Loads a model’s parameter dictionary
24
  network = network.to(device)
25
  network.eval()
 
19
  inference = parser.get_parsed_content("inferer")
20
  network = parser.get_parsed_content("network_def")
21
  preprocess = parser.get_parsed_content("preprocessing")
22
+ state_dict = torch.load("models/model.pt", map_location=torch.device('cpu'))
23
  network.load_state_dict(state_dict, strict=True) # Loads a model’s parameter dictionary
24
  network = network.to(device)
25
  network.eval()