Vignesh1997 commited on
Commit
7eae869
1 Parent(s): 3eab44f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -153,7 +153,7 @@ def find_algnments(
153
  bpe_target_map = []
154
  for i in translated_text.split():
155
  bpe_target_map += len(bert_tokenizer.tokenize(i)) * [i]
156
- st.success(source_embedding = bert_model(**source_tokens))
157
  source_embedding = bert_model(**source_tokens).hidden_states[8]
158
  target_embedding = bert_model(**target_tokens).hidden_states[8]
159
  target_embedding = target_embedding.transpose(-1, -2)
 
153
  bpe_target_map = []
154
  for i in translated_text.split():
155
  bpe_target_map += len(bert_tokenizer.tokenize(i)) * [i]
156
+ st.success(bert_model(**source_tokens))
157
  source_embedding = bert_model(**source_tokens).hidden_states[8]
158
  target_embedding = bert_model(**target_tokens).hidden_states[8]
159
  target_embedding = target_embedding.transpose(-1, -2)