add versions

sescore.py

@@ -122,8 +122,13 @@ class SEScore(evaluate.Metric):
         from huggingface_hub import snapshot_download
         # initialize roberta into str2encoder
         comet.encoders.str2encoder['RoBERTa'] = robertaEncoder
-        destination = snapshot_download(repo_id="xu1998hz/sescore_english_mt", revision="main")
-        self.scorer = load_from_checkpoint(f'{destination}/checkpoint/sescore_english_mt.ckpt')
+        if self.config_name == "default":
+            destination = snapshot_download(repo_id="xu1998hz/sescore_english_mt", revision="main")
+            self.scorer = load_from_checkpoint(f'{destination}/checkpoint/sescore_english_mt.ckpt')
+        else:
+            destination = snapshot_download(repo_id=self.config_name, revision="main")
+            suffix = self.config_name.split('/')[-1]
+            self.scorer = load_from_checkpoint(f'{destination}/checkpoint/{suffix}.ckpt')
 
     def _compute(self, predictions, references, gpus=None, progress_bar=False):
         if gpus is None: