Update app.py

app.py

@@ -24,16 +24,15 @@ def md_loading():
     # device = torch.device('cpu')
 
     tokenizer = XLMRobertaTokenizer.from_pretrained('xlm-roberta-large')
-    
     model = XLMRobertaForSequenceClassification.from_pretrained('xlm-roberta-large', num_labels=493)
     
     model_checkpoint = 'base3_44_ko.bin'
     project_path = './'
     output_model_file = os.path.join(project_path, model_checkpoint)
 
-#    model.load_state_dict(torch.load(output_model_file, map_location=torch.device('cpu')))
-    ckpt = torch.load(output_model_file, map_location=torch.device('cpu'))
-    model.load_state_dict(ckpt['model_state_dict'])
+    model.load_state_dict(torch.load(output_model_file, map_location=torch.device('cpu')))
+#    ckpt = torch.load(output_model_file, map_location=torch.device('cpu'))
+#    model.load_state_dict(ckpt['model_state_dict'])
     
 #    device = torch.device("cuda" if torch.cuda.is_available() and not False else "cpu")
 #    device = torch.device("cpu")