Handle string input in call method

Paulooh007 · Nov 5, 2023 · e43c5f6 · e43c5f6
1 parent cac6796
commit e43c5f6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/laser_encoders/models.py b/laser_encoders/models.py
@@ -97,6 +97,8 @@ def __init__(
     def __call__(self, text_or_batch):
         if self.spm_model:
             text_or_batch = self.tokenizer(text_or_batch)
+            if isinstance(text_or_batch, str):
+                text_or_batch = [text_or_batch]
             return self.encode_sentences(text_or_batch)
         else:
             raise ValueError(