Spaces:
Runtime error
Runtime error
Vignesh1997
commited on
Commit
•
3eab44f
1
Parent(s):
5049427
Update app.py
Browse files
app.py
CHANGED
@@ -153,6 +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 |
source_embedding = bert_model(**source_tokens).hidden_states[8]
|
157 |
target_embedding = bert_model(**target_tokens).hidden_states[8]
|
158 |
target_embedding = target_embedding.transpose(-1, -2)
|
@@ -211,8 +212,6 @@ if st.button("Run"):
|
|
211 |
)[0]
|
212 |
|
213 |
time_end = time.time()
|
214 |
-
st.success(user_input)
|
215 |
-
st.success(bert_model)
|
216 |
alignments = find_algnments(user_input, translated_text, bert_tokenizer, bert_model)
|
217 |
for i, j in alignments:
|
218 |
st.success(f"{i}->{j}")
|
|
|
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)
|
|
|
212 |
)[0]
|
213 |
|
214 |
time_end = time.time()
|
|
|
|
|
215 |
alignments = find_algnments(user_input, translated_text, bert_tokenizer, bert_model)
|
216 |
for i, j in alignments:
|
217 |
st.success(f"{i}->{j}")
|