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}
|