location-scoring-demo-google-api

Runtime error

jonathanjordan21 commited on Oct 10

Commit

571904c

verified ·

1 Parent(s): 400dcf8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -31,7 +31,7 @@ def negbinom_loss(y, mu, alpha):
     return -torch.mean(log_prob)
 model = NegBinomialModel(17)
-model.load_state_dict(torch.load("model_weights.pt"), map_location=torch.device('cpu'))
 model.eval()
 def predict_score(lat, lon):

     return -torch.mean(log_prob)
 model = NegBinomialModel(17)
+model.load_state_dict(torch.load("model_weights.pt", map_location='cpu'))
 model.eval()
 def predict_score(lat, lon):