Update app.py

app.py

@@ -177,7 +177,8 @@ def find_algnments(
     return list(set(align_words))
 
 
-st.title("M2M100 Translator")
+st.title("Multilingual alignment")
+st.write("The pretranined models used to make the above setup are ")
 st.write(
     "M2M100 is a multilingual encoder-decoder (seq-to-seq) model trained for Many-to-Many multilingual translation. It was introduced in this paper https://arxiv.org/abs/2010.11125 and first released in https://github.com/pytorch/fairseq/tree/master/examples/m2m_100 repository. The model that can directly translate between the 9,900 directions of 100 languages.\n"
 )
@@ -216,7 +217,7 @@ if st.button("Run"):
 
     time_end = time.time()
     alignments = find_algnments(user_input, translated_text, bert_tokenizer, bert_model)
-    for i, j in alignments:
-        st.success(f"{i}->{j}")
+    for string in alignments:
+        st.success(string)
 
     st.write(f"Computation time: {round((time_end-time_start),3)} sec")