Spaces:
Sleeping
Sleeping
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
|