Update app.py

app.py

@@ -25,7 +25,7 @@ def highlight_diff(sent, sent_main):
     new_toks = []
     for i, (tok, tok_main) in enumerate(zip(tokens, tokens_main)):
         if tok != tok_main:
-            new_toks.append('$\textcolor{red}{***' + tok + '***}$')
+            new_toks.append('***' + tok + '***')
         else:
             new_toks.append(tok)
     
@@ -91,7 +91,7 @@ def get_best_hypotheses(sentence, beam_size, max_steps, epsilon=1e-3, pretty_out
     
     st.write(f"step #0:")
     st.write(f"-- 1: (positive probability ~ {round(current_beam[sentence], 5)})")
-    st.markdown(f"$\qquad${sentence}")
+    st.write(f"$\qquad${sentence}")
     
     for step in range(max_steps):
         current_beam, used_pos = beam_get_replacements(current_beam, beam_size, epsilon, used_poss)
@@ -99,7 +99,7 @@ def get_best_hypotheses(sentence, beam_size, max_steps, epsilon=1e-3, pretty_out
         st.write(f"\nstep #{step+1}:")
         for i, (sent, prob) in enumerate(current_beam.items()):
             st.write(f"-- {i+1}: (positive probability ~ {round(prob, 5)})")
-            st.markdown(f"$\qquad${highlight_diff(sent, sentence) if pretty_output else sent}")
+            st.write(f"$\qquad${highlight_diff(sent, sentence) if pretty_output else sent}")
         
         if used_pos is None:
             return current_beam, used_poss