Spaces:
Running
Running
Update lib/comparison.py
Browse files- lib/comparison.py +1 -1
lib/comparison.py
CHANGED
@@ -4,7 +4,7 @@ from sklearn.metrics.pairwise import cosine_similarity
|
|
4 |
import numpy as np
|
5 |
|
6 |
# Load BERT tokenizer and model
|
7 |
-
bert_model_name = "yiyanghkust/finbert-tone" #"bert-base-uncased"
|
8 |
tokenizer = BertTokenizer.from_pretrained(bert_model_name)
|
9 |
model = BertModel.from_pretrained(bert_model_name)
|
10 |
model.eval() # Set to evaluation mode
|
|
|
4 |
import numpy as np
|
5 |
|
6 |
# Load BERT tokenizer and model
|
7 |
+
bert_model_name = "bert-base-uncased"#"yiyanghkust/finbert-tone" #"bert-base-uncased"
|
8 |
tokenizer = BertTokenizer.from_pretrained(bert_model_name)
|
9 |
model = BertModel.from_pretrained(bert_model_name)
|
10 |
model.eval() # Set to evaluation mode
|