update app.py

app.py

@@ -74,10 +74,10 @@ class inferSSCL():
         self.base_models['embedding'].weight = torch.nn.Parameter(emb_para)
         
         self.base_models['asp_weight'] = torch.nn.Linear(emb_size, self.batch_data['n_aspects']).to(device)
-        self.base_models['asp_weight'].load_state_dict(torch.load('./asp_weight.model'))
+        self.base_models['asp_weight'].load_state_dict(torch.load('./asp_weight.model', map_location=torch.device('cpu')))
         
         self.base_models['attn_kernel'] = torch.nn.Linear(emb_size, emb_size).to(device)
-        self.base_models['attn_kernel'].load_state_dict(torch.load('./attn_kernel.model'), strict=False)
+        self.base_models['attn_kernel'].load_state_dict(torch.load('./attn_kernel.model', map_location=torch.device('cpu')), strict=False)
         
     
     def build_pipe(self):