Spaces:
Build error
Build error
remove src
Browse files- sescore.py +1 -1
sescore.py
CHANGED
@@ -134,7 +134,7 @@ class SEScore(evaluate.Metric):
|
|
134 |
destination = snapshot_download(repo_id="xu1998hz/sescore_english_mt", revision="main")
|
135 |
self.scorer = load_from_checkpoint(f'{destination}/checkpoint/sescore_english_mt.ckpt')
|
136 |
|
137 |
-
def _compute(self,
|
138 |
if gpus is None:
|
139 |
gpus = 1 if torch.cuda.is_available() else 0
|
140 |
data = {"src": references, "mt": predictions}
|
|
|
134 |
destination = snapshot_download(repo_id="xu1998hz/sescore_english_mt", revision="main")
|
135 |
self.scorer = load_from_checkpoint(f'{destination}/checkpoint/sescore_english_mt.ckpt')
|
136 |
|
137 |
+
def _compute(self, predictions, references, gpus=None, progress_bar=False):
|
138 |
if gpus is None:
|
139 |
gpus = 1 if torch.cuda.is_available() else 0
|
140 |
data = {"src": references, "mt": predictions}
|