Upload app.py

app.py

@@ -13,7 +13,7 @@ import  torch_geometric.transforms as T
 from torch_geometric.nn import LightGCN
 import utils
 
-device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+device = torch.device('cpu')
 data = torch.load("processed_MVL_light.pt")
 ch = torch.load('./lightGCNModel_num_layers_MovieLens100K_checkpoint.pt')
 lightGCNModel = LightGCN(
@@ -21,11 +21,12 @@ lightGCNModel = LightGCN(
         embedding_dim=64,
         num_layers=3,
     ).to(device)
-optimizer = torch.optim.Adam(lightGCNModel.parameters(), lr=0.005)
+
+# optimizer = torch.optim.Adam(lightGCNModel.parameters(), lr=0.005)
 mask_train = data.edge_index[0] < data.edge_index[1]
 train_edge_label_index = data.edge_index[:, mask_train]
 lightGCNModel.load_state_dict(ch['model_state_dict'])
-optimizer.load_state_dict(ch['optimizer_state_dict'])
+# optimizer.load_state_dict(ch['optimizer_state_dict'])
 num_items = 1682
 num_users = 943