Changement de traducteur en seq2seq

requirements.txt

@@ -24,7 +24,6 @@ sentencepiece==0.1.99
 openai-whisper==20231117
 sounddevice==0.4.6
 torch==2.2.0
-translate==3.6.1
 speechrecognition==3.10.1
 audio_recorder_streamlit==0.0.8
 whisper==1.1.10

tabs/modelisation_seq2seq_tab.py

@@ -4,7 +4,8 @@ import numpy as np
 import os
 from sacrebleu import corpus_bleu
 from transformers import pipeline
-from translate import Translator
+# from translate import Translator
+from deep_translator import GoogleTranslator
 from audio_recorder_streamlit import audio_recorder
 import speech_recognition as sr
 import whisper
@@ -461,7 +462,8 @@ def run():
         with col2:
             st.write(":red[**Trad. Google Translate**]")
             try:
-                translator = Translator(to_lang=l_tgt, from_lang=Lang_detected)
+                # translator = Translator(to_lang=l_tgt, from_lang=Lang_detected)
+                translator = GoogleTranslator(source=Lang_detected, target=l_tgt)
                 if custom_sentence!="":
                     translation = translator.translate(custom_sentence)
                     st.write("**"+l_tgt+" :**  "+translation)
@@ -518,7 +520,8 @@ def run():
                     st.write("")
                     st.write("**"+Lang_detected+" :**  :blue["+custom_sentence+"]")
                     st.write("")
-                    translator = Translator(to_lang=l_tgt, from_lang=Lang_detected)
+                    # translator = Translator(to_lang=l_tgt, from_lang=Lang_detected)
+                    translator = GoogleTranslator(source=Lang_detected, target=l_tgt)
                     translation = translator.translate(custom_sentence)
                     st.write("**"+l_tgt+" :**  "+translation)
                     st.write("")