Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 import pandas as pd
 import numpy as np
+import re
 
 import torch
 from transformers import pipeline
@@ -36,4 +37,6 @@ with st.form("my_form"):
 
    
    if submitted:
-       st.write(pipe(text, num_return_sequences=1, max_length=100)[0]["generated_text"])
+       text = pipe(text, num_return_sequences=1, max_length=200, repetition_penalty=2.0)[0]["generated_text"]
+       text = re.sub("( [A-Z])", r"\n\1", text)
+       st.write(text)