Update app.py

app.py

@@ -6,7 +6,7 @@ from datetime import datetime
 
 def to_md(text):
     # return text.replace("\n", "<br />")
-    return text
+    return text.replace("\n", "<br />")
 
 @st.cache
 def infer(
@@ -18,7 +18,7 @@ def infer(
     top_k=40,
     num_completions=1,
     seed=42,
-    stop="."
+    stop="\n"
 ):
     model_name_map = {
         "GPT-JT-6B-v1": "Together-gpt-JT-6B-v1",
@@ -130,7 +130,7 @@ def main():
         top_p = st.text_input('top_p', st.session_state.top_p)
         # num_completions = st.text_input('num_completions (only the best one will be returend)', "1")
         num_completions = "1"
-        stop = st.text_input('stop, split by;', r'.')
+        stop = st.text_input('stop, split by;', r'\n')
         # seed = st.text_input('seed', "42")
         seed = "42"
 
@@ -163,7 +163,6 @@ 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+"'''"),
             )
-            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__':