Blanca commited on
Commit
22137d4
·
verified ·
1 Parent(s): 739a877

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -17,7 +17,7 @@ from sentence_transformers import SentenceTransformer
17
  from scorer import question_scorer
18
  from content import format_error, format_warning, format_log, TITLE, INTRODUCTION_TEXT, SUBMISSION_TEXT, CITATION_BUTTON_LABEL, CITATION_BUTTON_TEXT, model_hyperlink
19
 
20
- model = SentenceTransformer("stsb-mpnet-base-v2")
21
 
22
  TOKEN = os.environ.get("TOKEN", None)
23
 
@@ -167,9 +167,9 @@ def add_new_eval(
167
  cq_text = cq['cq']
168
 
169
  #if args.metric == 'similarity':
170
- sentence_embedding = model.encode(cq_text)
171
- reference_embedding = model.encode(reference_set)
172
- sims = model.similarity(sentence_embedding, reference_embedding).tolist()[0]
173
 
174
  winner = np.argmax(sims)
175
  # make sure the similarity of the winning reference sentence is at least 0.65
 
17
  from scorer import question_scorer
18
  from content import format_error, format_warning, format_log, TITLE, INTRODUCTION_TEXT, SUBMISSION_TEXT, CITATION_BUTTON_LABEL, CITATION_BUTTON_TEXT, model_hyperlink
19
 
20
+ similarity_model = SentenceTransformer("stsb-mpnet-base-v2")
21
 
22
  TOKEN = os.environ.get("TOKEN", None)
23
 
 
167
  cq_text = cq['cq']
168
 
169
  #if args.metric == 'similarity':
170
+ sentence_embedding = similarity_model.encode(cq_text)
171
+ reference_embedding = similarity_model.encode(reference_set)
172
+ sims = similarity_model.similarity(sentence_embedding, reference_embedding).tolist()[0]
173
 
174
  winner = np.argmax(sims)
175
  # make sure the similarity of the winning reference sentence is at least 0.65