Update modelisation_seq2seq_tab.py

tabs/modelisation_seq2seq_tab.py

@@ -96,11 +96,12 @@ def fetch_translation(url):
 def display_translation(n1, Lang, model_type):
     global df_data_src, df_data_tgt, placeholder
     
+    n = 3
     placeholder = st.empty()
     with st.status(":sunglasses:", expanded=True):
-        s = df_data_src.iloc[n1:n1+3][0].tolist()
+        s = df_data_src.iloc[n1:n1+n][0].tolist()
         s_trad = []
-        s_trad_ref = df_data_tgt.iloc[n1:n1+3][0].tolist()
+        s_trad_ref = df_data_tgt.iloc[n1:n1+n][0].tolist()
         source = Lang[:2]
         target = Lang[-2:]
         params = []
@@ -108,12 +109,12 @@ def display_translation(n1, Lang, model_type):
             url_base = "https://demosthene-or-api-avr23-cds-translation.hf.space/small_vocab/rnn"
         else:
             url_base = "https://demosthene-or-api-avr23-cds-translation.hf.space/small_vocab/transformer"
-        for i in range(3):
+        for i in range(n):
             # Construction de l'URL avec les paramètres
             url = f"{url_base}?lang_tgt={target}&texte={s[i]}"
             params.append(url)
         
-        with Pool(3) as p:
+        with Pool(n) as p:
             # Appels parallèles à fetch_translation avec les URLs
             responses = p.map(fetch_translation, params)