Cachoups commited on
Commit
6d183c6
·
verified ·
1 Parent(s): 74b6936

Update lib/comparison.py

Browse files
Files changed (1) hide show
  1. 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