Update app.py

app.py

@@ -28,10 +28,10 @@ class BertModel(torch.nn.Module):
         return output
 
 model_preamble = BertModel()
-model_preamble = torch.load("nerbert_preamble.pt")
+model_preamble = torch.load("nerbert_preamble.pt", map_location=torch.device('cpu'))
 
 model_judgment = BertModel()
-model_judgment = torch.load("nerbert.pt")
+model_judgment = torch.load("nerbert.pt", map_location=torch.device('cpu'))
 
 
 unique_labels_preamble = {'I-PETITIONER', 'I-COURT', 'B-COURT', 'B-JUDGE', 'I-LAWYER', 'B-RESPONDENT', 'I-JUDGE', 'B-PETITIONER', 'I-RESPONDENT', 'B-LAWYER', 'O'}