Update app.py

app.py

@@ -153,6 +153,7 @@ def find_algnments(
     bpe_target_map = []
     for i in translated_text.split():
         bpe_target_map += len(bert_tokenizer.tokenize(i)) * [i]
+    st.success(source_embedding = bert_model(**source_tokens))    
     source_embedding = bert_model(**source_tokens).hidden_states[8]
     target_embedding = bert_model(**target_tokens).hidden_states[8]
     target_embedding = target_embedding.transpose(-1, -2)
@@ -211,8 +212,6 @@ if st.button("Run"):
         )[0]
 
     time_end = time.time()
-    st.success(user_input)
-    st.success(bert_model)
     alignments = find_algnments(user_input, translated_text, bert_tokenizer, bert_model)
     for i, j in alignments:
         st.success(f"{i}->{j}")