Spaces:
Runtime error
Runtime error
Vignesh1997
commited on
Commit
•
7eae869
1
Parent(s):
3eab44f
Update app.py
Browse files
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(
|
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)
|