Spaces:
Running
Running
José Ángel González
commited on
Commit
·
e315877
1
Parent(s):
7004eb5
add nltk punkt
Browse files- generation_evaluator.py +4 -0
generation_evaluator.py
CHANGED
|
@@ -4,6 +4,7 @@ import numpy as np
|
|
| 4 |
import spacy
|
| 5 |
import torch
|
| 6 |
from alignscore import AlignScore
|
|
|
|
| 7 |
|
| 8 |
_CITATION = """\
|
| 9 |
@inproceedings{lin-2004-rouge,
|
|
@@ -148,6 +149,9 @@ class GenerationEvaluator(evaluate.Metric):
|
|
| 148 |
except OSError:
|
| 149 |
spacy.cli.download("en_core_web_sm")
|
| 150 |
|
|
|
|
|
|
|
|
|
|
| 151 |
# Download AlignScore model and move to GPU if possible
|
| 152 |
model_path = dl_manager.download(ALIGNSCORE_ARGS["ckpt_path"])
|
| 153 |
ALIGNSCORE_ARGS["ckpt_path"] = model_path
|
|
|
|
| 4 |
import spacy
|
| 5 |
import torch
|
| 6 |
from alignscore import AlignScore
|
| 7 |
+
import nltk
|
| 8 |
|
| 9 |
_CITATION = """\
|
| 10 |
@inproceedings{lin-2004-rouge,
|
|
|
|
| 149 |
except OSError:
|
| 150 |
spacy.cli.download("en_core_web_sm")
|
| 151 |
|
| 152 |
+
# Download punkt for AlignScore
|
| 153 |
+
nltk.download('punkt_tab')
|
| 154 |
+
|
| 155 |
# Download AlignScore model and move to GPU if possible
|
| 156 |
model_path = dl_manager.download(ALIGNSCORE_ARGS["ckpt_path"])
|
| 157 |
ALIGNSCORE_ARGS["ckpt_path"] = model_path
|