Upload app.py

app.py

@@ -14,8 +14,8 @@ from torch_geometric.nn import LightGCN
 import utils
 
 device = torch.device('cpu')
-data = torch.load("processed_MVL_light.pt")
-ch = torch.load('./lightGCNModel_num_layers_MovieLens100K_checkpoint.pt')
+data = torch.load("processed_MVL_light.pt", map_location=torch.device('cpu'))
+ch = torch.load('./lightGCNModel_num_layers_MovieLens100K_checkpoint.pt', map_location=torch.device('cpu'))
 lightGCNModel = LightGCN(
         num_nodes=data.num_nodes,
         embedding_dim=64,