well done now

utils/utils_function.py

@@ -42,7 +42,13 @@ class Translator:
         # Encodage du texte en tokens
         tokens = self.model.encode(text)
         
+        # Utilisation de la méthode generate avec le paramètre beam
+        translations = self.model.generate(tokens, beam=5)
+        print(type(translations))
+        print(translations[0])
+        best_translation_tokens = [translations[i]['tokens'].tolist() for i in range(5)]
+
         # Décodage des tokens en traduction
-        translation = self.model.decode(tokens)
+        translations = [self.model.decode(best_translation_tokens[i]) for i in range(5)]
         
-        return translation
+        return "\n".join(translations)