Update app.py

app.py

@@ -21,12 +21,10 @@ st.header('한국표준산업분류 자동코딩 서비스')
 @st.experimental_memo(max_entries=20)
 def md_loading():
     ## cpu
-    # device = torch.device('cpu')
-
     device = torch.device("cpu")
 
-    tokenizer = XLMRobertaTokenizer.from_pretrained('xlm-roberta-large')
-    model = XLMRobertaForSequenceClassification.from_pretrained('xlm-roberta-large', num_labels=493)
+    tokenizer = XLMRobertaTokenizer.from_pretrained('xlm-roberta-base')
+    model = XLMRobertaForSequenceClassification.from_pretrained('xlm-roberta-base', num_labels=493)
     
     model_checkpoint = 'base3_44_ko.bin'
     project_path = './'