change model weight save path

app.py

@@ -40,8 +40,9 @@ model.fc = nn.Sequential(
     nn.Linear(model.fc.in_features, 2)
 )
 
+device = torch.device('cpu')
 urllib.request.urlretrieve(MODEL_URL, MODEL_PATH)
-model.load_state_dict(torch.load(MODEL_PATH))
+model.load_state_dict(torch.load(MODEL_PATH, map_location=device))
 model.eval()
 for _, p in model.named_parameters():
     p.requires_grad = False