any0019 commited on
Commit
36666ce
·
1 Parent(s): b78c6b8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -25,7 +25,7 @@ def highlight_diff(sent, sent_main):
25
  new_toks = []
26
  for i, (tok, tok_main) in enumerate(zip(tokens, tokens_main)):
27
  if tok != tok_main:
28
- new_toks.append(f'\textcolor\{red\}\{**{tok}**\}')
29
  else:
30
  new_toks.append(tok)
31
 
@@ -115,7 +115,7 @@ default_value = "write your review here (in lower case - vocab reasons)"
115
  sentence = st.text_area("Text", default_value, height = 275)
116
  beam_size = st.sidebar.slider("Beam size", value = 3, min_value = 1, max_value=20, step=1)
117
  max_steps = st.sidebar.slider("Max steps", value = 3, min_value = 1, max_value=10, step=1)
118
- # prettyfy = st.sidebar.slider("Higlight changes", value = 0, min_value = 0, max_value=1, step=1)
119
 
120
- # beam, used_poss = get_best_hypotheses(sentence, beam_size=beam_size, max_steps=max_steps, pretty_output=bool(prettyfy))
121
- beam, used_poss = get_best_hypotheses(sentence, beam_size=beam_size, max_steps=max_steps, pretty_output=False)
 
25
  new_toks = []
26
  for i, (tok, tok_main) in enumerate(zip(tokens, tokens_main)):
27
  if tok != tok_main:
28
+ new_toks.append('\textcolor\{red\}\{**{}**\}'.format(tok))
29
  else:
30
  new_toks.append(tok)
31
 
 
115
  sentence = st.text_area("Text", default_value, height = 275)
116
  beam_size = st.sidebar.slider("Beam size", value = 3, min_value = 1, max_value=20, step=1)
117
  max_steps = st.sidebar.slider("Max steps", value = 3, min_value = 1, max_value=10, step=1)
118
+ prettyfy = st.sidebar.slider("Higlight changes", value = 0, min_value = 0, max_value=1, step=1)
119
 
120
+ beam, used_poss = get_best_hypotheses(sentence, beam_size=beam_size, max_steps=max_steps, pretty_output=bool(prettyfy))
121
+ # beam, used_poss = get_best_hypotheses(sentence, beam_size=beam_size, max_steps=max_steps, pretty_output=False)