Spaces:
Sleeping
Sleeping
HalteroXHunter
commited on
Commit
·
d5750c7
1
Parent(s):
c418edf
small fixes
Browse files- app.py +1 -1
- generation_evaluator.py +1 -1
- requirements.txt +2 -0
app.py
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
import evaluate
|
2 |
from gradio_tst import launch_gradio_widget2
|
3 |
|
4 |
-
module = evaluate.load("
|
5 |
launch_gradio_widget2(module)
|
|
|
1 |
import evaluate
|
2 |
from gradio_tst import launch_gradio_widget2
|
3 |
|
4 |
+
module = evaluate.load("generation_evaluator.py")
|
5 |
launch_gradio_widget2(module)
|
generation_evaluator.py
CHANGED
@@ -159,7 +159,7 @@ class GenerationEvaluator(evaluate.Metric):
|
|
159 |
predictions=predictions, references=references
|
160 |
)
|
161 |
|
162 |
-
bert_score = evaluate.load("
|
163 |
bert_score_results = bert_score.compute(
|
164 |
predictions=predictions, references=references,
|
165 |
lang="en"
|
|
|
159 |
predictions=predictions, references=references
|
160 |
)
|
161 |
|
162 |
+
bert_score = evaluate.load("bertscore")
|
163 |
bert_score_results = bert_score.compute(
|
164 |
predictions=predictions, references=references,
|
165 |
lang="en"
|
requirements.txt
CHANGED
@@ -2,3 +2,5 @@ evaluate
|
|
2 |
datasets
|
3 |
scikit-learn
|
4 |
gradio
|
|
|
|
|
|
2 |
datasets
|
3 |
scikit-learn
|
4 |
gradio
|
5 |
+
bert_score
|
6 |
+
git+https://github.com/google-research/bleurt.git
|