diff --git a/turjuman/turjuman.py b/turjuman/turjuman.py index f9e0afb..dd7d624 100644 --- a/turjuman/turjuman.py +++ b/turjuman/turjuman.py @@ -59,7 +59,7 @@ def translate(self, sources, search_method, seq_length=300, max_outputs=1, num_b generated_text = self.tokenizer.batch_decode(outputs, skip_special_tokens=True) - return outputs + return generated_text def translate_from_file(self, input_file, search_method, seq_length=300, max_outputs=1, num_beams=5, no_repeat_ngram_size=2, top_p=0.95, top_k=50, batch_size=25):