Update app.py

app.py

@@ -90,7 +90,7 @@ def random_properties():
 def load_model(model_choice):
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     model = load_graph_decoder(path=model_choice)
-    model.to(device)
+    # model.to(device)
     print('in load_model', torch.device("cuda" if torch.cuda.is_available() else "cpu"))
     return (model, device)