Update app.py

app.py

@@ -247,7 +247,7 @@ class Tester(TesterBase):
 
     def load_model(self, model_path):
         self.model = torch.nn.DataParallel(self.model)
-        cpk = torch.load(model_path)
+        cpk = torch.load(model_path, map_location=torch.device('cpu'))
         saved_state_dict = cpk['model']
         self.model.load_state_dict(saved_state_dict)
         self.model = self.model.module