fix translation (in array)

app.py

@@ -19,9 +19,10 @@ if st.button('générer'):
     st.write('response is :', responseBase)
 
     st.write('Part 2')
-    completeResponse = englishtofrench(longText(responseBase, question))
-    st.write('en : ',longText(responseBase, question))
-    st.write('fr : ',completeResponse)
+    enResponse = longText(responseBase, question)
+    frResponse = englishtofrench(enResponse)
+    st.write('en : ',enResponse)
+    st.write('fr : ',frResponse)
 
     st.write('Part 3')
     st.write('Next step : TTS')

translation.py

@@ -11,11 +11,11 @@ def frenchtoenglish(input):
     tokenizer.src_lang = "fr"
     encoded_fr = tokenizer(input, return_tensors="pt")
     generated_tokens = model.generate(**encoded_fr, forced_bos_token_id=tokenizer.get_lang_id("en"))
-    return tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
+    return tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]
 
 def englishtofrench(input):
     # translate English to French
     tokenizer.src_lang = "en"
     encoded_en = tokenizer(input, return_tensors="pt")
     generated_tokens = model.generate(**encoded_en, forced_bos_token_id=tokenizer.get_lang_id("fr"))
-    return tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
+    return tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]