HalteroXHunter commited on
Commit
d5750c7
·
1 Parent(s): c418edf

small fixes

Browse files
Files changed (3) hide show
  1. app.py +1 -1
  2. generation_evaluator.py +1 -1
  3. 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("textgen_evaluator.py")
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("bert_score")
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