Update app.py

app.py

@@ -6,8 +6,9 @@ import numpy as np
 
 # Load the model and tokenizer
 def load_model():
-    tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
-    model = BertForSequenceClassification.from_pretrained('bert-base-uncased', num_labels=7)
+    tokenizer = BertTokenizer.from_pretrained('beomi/kcbert-base')
+    config = BertConfig.from_pretrained('beomi/kcbert-base', num_labels=7)
+    model = BertForSequenceClassification.from_pretrained('beomi/kcbert-base', config=config)
     model_state_dict = torch.load('sentiment7_model_acc8878.pth', map_location=torch.device('cpu')) # cpu 사용
     model.load_state_dict(model_state_dict)
     model.eval()