xu1998hz commited on
Commit
127b211
·
1 Parent(s): 0a11401

update weight loc

Browse files
Files changed (1) hide show
  1. sescore.py +3 -3
sescore.py CHANGED
@@ -88,11 +88,11 @@ class SEScore(evaluate.Metric):
88
  import gdown
89
  import os
90
  url = "https://drive.google.com/uc?id=1QgMP_Y4QCbvDMTeVacYt0J76OYvwWK9V&export=download&confirm=true"
91
- output = 'sescore_download.gz'
92
  gdown.download(url, output, quiet=False)
93
- cmd = 'tar -xvf sescore_download.gz'
94
  os.system(cmd)
95
- self.scorer = load_from_checkpoint('sescore_download/zh_en/checkpoint/sescore_english.ckpt')
96
 
97
  def _compute(self, sources, predictions, references, gpus=None, progress_bar=False):
98
  if gpus is None:
 
88
  import gdown
89
  import os
90
  url = "https://drive.google.com/uc?id=1QgMP_Y4QCbvDMTeVacYt0J76OYvwWK9V&export=download&confirm=true"
91
+ output = 'sescore_ckpt.gz'
92
  gdown.download(url, output, quiet=False)
93
+ cmd = 'tar -xvf sescore_ckpt.gz'
94
  os.system(cmd)
95
+ self.scorer = load_from_checkpoint('sescore_ckpt/zh_en/checkpoint/sescore_english.ckpt')
96
 
97
  def _compute(self, sources, predictions, references, gpus=None, progress_bar=False):
98
  if gpus is None: