Update main_dl.py

main_dl.py

@@ -241,14 +241,14 @@ rnn_en_fr, rnn_fr_en, transformer_en_fr, transformer_fr_en = load_all_data()
 async def display_translation(n1, Lang,model_type):
     global df_data_src, df_data_tgt, placeholder
     
-    async def decode_seq_all(model_type,s,):
+    async def decode_seq_all(model_type,s, source, target):
             for i in range(3):
                 if model_type==1:
                     s_trad.append(decode_sequence_rnn(s[i], source, target))
                 else:
                     s_trad.append(decode_sequence_tranf(s[i], source, target))
             await (len(s_trad) == 3)
-            return 
+            return s_trad
     
     placeholder = st.empty()
     with st.status(":sunglasses:", expanded=True):
@@ -257,7 +257,7 @@ async def display_translation(n1, Lang,model_type):
         s_trad_ref = df_data_tgt.iloc[n1:n1+5][0].tolist()
         source = Lang[:2]
         target = Lang[-2:]
-        await decode_seq_all()
+        await decode_seq_all(model_type,s, source, target)
         for i in range(3):
             st.write("**"+source+"   :**  :blue["+ s[i]+"]")
             st.write("**"+target+"   :**  "+s_trad[-1])