Update app.py

app.py

@@ -33,7 +33,7 @@ def pred(dist,speed,accel,timedelta,jerk,bearing,bearing_rate):
     n_layers = 2
     drop_prob = 0.2
     net = ModalityLSTM(trip_dim, output_size, batch_size, hidden_dim, n_layers, train_on_gpu, drop_prob, lstm_drop_prob=0.2)
-    net.load_state_dict(torch.load("Model_Wieghts"),map_location ='cpu')
+    net.load_state_dict(torch.load("Model_Wieghts"),map_location = device)
     net.eval()
 
     a=torch.tensor([[dist,speed,accel,timedelta,jerk,bearing,bearing_rate]])