Spaces:
Runtime error
Runtime error
agueroooooooooo
commited on
Commit
•
f62ce63
1
Parent(s):
148ae5d
Update app.py
Browse filesadding map location
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]])
|