Update app.py
Browse files
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 |
-
|
| 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 =
|
| 171 |
-
reference_embedding =
|
| 172 |
-
sims =
|
| 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
|