Update modelisation_seq2seq_tab.py

tabs/modelisation_seq2seq_tab.py

@@ -89,12 +89,11 @@ def display_translation(n1, Lang,model_type):
         st.write("<p style='text-align:center;background-color:red; color:white')>Score Bleu = "+str(int(round(corpus_bleu(s_trad,[s_trad_ref]).score,0)))+"%</p>", \
             unsafe_allow_html=True)
 '''
-def fetch_translation(param):
-    print("->"+param)
-    return requests.get(url=param)
-    
- 
-def display_translation(n1, Lang,model_type):
+def fetch_translation(url):
+    print("->"+url)
+    return requests.get(url)
+
+def display_translation(n1, Lang, model_type):
     global df_data_src, df_data_tgt, placeholder
     
     placeholder = st.empty()
@@ -105,24 +104,25 @@ def display_translation(n1, Lang,model_type):
         source = Lang[:2]
         target = Lang[-2:]
         params = []
-        if model_type==1:
-            # URL de votre endpoint FastAPI avec les paramètres de requête
-            url = "https://demosthene-or-api-avr23-cds-translation.hf.space/small_vocab/rnn"
+        if model_type == 1:
+            url_base = "https://demosthene-or-api-avr23-cds-translation.hf.space/small_vocab/rnn"
         else:
-            # URL de votre endpoint FastAPI avec les paramètres de requête
-            url = "https://demosthene-or-api-avr23-cds-translation.hf.space/small_vocab/transformer"
+            url_base = "https://demosthene-or-api-avr23-cds-translation.hf.space/small_vocab/transformer"
         for i in range(3):
-            # Ajout du parametre à la liste des parametres
-            params.append(url+"?lang_tgt="+target+"&texte="+s[i])
+            # 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:
+            # Appels parallèles à fetch_translation avec les URLs
             responses = p.map(fetch_translation, params)
             
-            for i in range(3):
-                s_trad.append(responses[i].json())
-                st.write("**"+source+"   :**  :blue["+ s[i]+"]")
-                st.write("**"+target+"   :**  "+s_trad[-1])
-                st.write("**ref. :** "+s_trad_ref[i])
+            for response, s_text in zip(responses, s):
+                # Récupération des résultats
+                s_trad.append(response.json())
+                st.write(f"**{source}   :**  :blue[{s_text}]")
+                st.write(f"**{target}   :**  {s_trad[-1]}")
+                st.write(f"**ref. :** {s_trad_ref[i]}")
                 st.write("")
     with placeholder:
         st.write("<p style='text-align:center;background-color:red; color:white')>Score Bleu = "+str(int(round(corpus_bleu(s_trad,[s_trad_ref]).score,0)))+"%</p>", \