jogonba2 commited on
Commit
e315877
·
1 Parent(s): 7004eb5

add nltk punkt

Browse files
Files changed (1) hide show
  1. 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