Update app.py

app.py

@@ -1,5 +1,3 @@
-
-
 import streamlit as st
 import requests
 import time
@@ -7,7 +5,8 @@ from ast import literal_eval
 from datetime import datetime
 
 def to_md(text):
-    return text.replace("\n", "<br />")
+    # return text.replace("\n", "<br />")
+    return text
 
 @st.cache
 def infer(
@@ -35,7 +34,7 @@ def infer(
     assert 1 <= num_completions <= 5
     assert 0.0 <= temperature <= 10.0
     assert 0.0 <= top_p <= 1.0
-    
+
     if temperature == 0.0:
         temperature = 0.01
 
@@ -164,7 +163,8 @@ def main():
                 prompt, model_name=model_name, max_new_tokens=max_new_tokens, temperature=temperature, top_p=top_p, top_k=top_k,
                 num_completions=num_completions, seed=seed, stop=literal_eval("'''"+stop+"'''"),
             )
-            generated_area.markdown("<b>" + to_md(prompt) + "</b>" + to_md(report_text), unsafe_allow_html=True)
+            print(report_text)
+            generated_area.markdown("<b>" + to_md(prompt) + "</b><mark style='background-color: #cbeacd'>" + to_md(report_text)+"</mark>", unsafe_allow_html=True)
         
 if __name__ == '__main__':
     main()