Update README.md

README.md

@@ -71,8 +71,8 @@ def cal_score(a, b):
     return torch.mm(a_norm, b_norm.transpose(0, 1)) * 100 
 
 # Load pre-trained model
-model = SimCSEModel.from_pretrained("daekeun-ml/KoSimCSE-unsupervised-roberta-large")
-tokenizer = AutoTokenizer.from_pretrained("daekeun-ml/KoSimCSE-unsupervised-roberta-large")
+model = SimCSEModel.from_pretrained("daekeun-ml/KoSimCSE-unsupervised-roberta-base")
+tokenizer = AutoTokenizer.from_pretrained("daekeun-ml/KoSimCSE-unsupervised-roberta-base")
 
 # Inference example
 sentences = ['이번 주 일요일에 분당 이마트 점은 문을 여나요?',