agueroooooooooo commited on
Commit
f62ce63
1 Parent(s): 148ae5d

Update app.py

Browse files

adding map location

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ def pred(dist,speed,accel,timedelta,jerk,bearing,bearing_rate):
33
  n_layers = 2
34
  drop_prob = 0.2
35
  net = ModalityLSTM(trip_dim, output_size, batch_size, hidden_dim, n_layers, train_on_gpu, drop_prob, lstm_drop_prob=0.2)
36
- net.load_state_dict(torch.load("Model_Wieghts"))
37
  net.eval()
38
 
39
  a=torch.tensor([[dist,speed,accel,timedelta,jerk,bearing,bearing_rate]])
 
33
  n_layers = 2
34
  drop_prob = 0.2
35
  net = ModalityLSTM(trip_dim, output_size, batch_size, hidden_dim, n_layers, train_on_gpu, drop_prob, lstm_drop_prob=0.2)
36
+ net.load_state_dict(torch.load("Model_Wieghts"),map_location ='cpu')
37
  net.eval()
38
 
39
  a=torch.tensor([[dist,speed,accel,timedelta,jerk,bearing,bearing_rate]])