change location

model.py

@@ -31,6 +31,6 @@ class EnergySmellsDetector(nn.Module):
 
         # Load model
         model = EnergySmellsDetector(PRETRAINED_MODEL)
-        model.load_state_dict(torch.load(model_path))
+        model.load_state_dict(torch.load(model_path, map_location=torch.device('cpu')))
 
         return model