allow inference of hyperpcm on cpu

app.py

@@ -143,7 +143,7 @@ def retrieval():
         memory = dataset
         model = HyperPCM(memory=memory).to(device)
         model = torch.nn.DataParallel(model)
-        model.load_state_dict(torch.load(checkpoint_path))
+        model.load_state_dict(torch.load(checkpoint_path, map_location=lambda storage, loc: storage))
         model.eval()
 
         with torch.set_grad_enabled(False):