Added functionality

app.py

@@ -17,7 +17,7 @@ from transformers import pipeline, set_seed
 
 generator = pipeline('text-generation', model='openai-gpt')
 
-def generate(initial_text, length=10, return_sequences=1):
+def generate(initial_text, length=50, return_sequences=1):
     set_seed(randint(1,1000))
     result = generator(initial_text, max_length = length, num_return_sequences = return_sequences)
     return result[0]["generated_text"]
@@ -33,17 +33,13 @@ def type_text(text):
 #text = input("Enter something to begin with... ")
 #print(".\n.\n.\nGenerating\n.\n.\n.")
 
-text = st.text_input('Enter something to begin with...', placeholder='I looked at her, then i realized...', key="value")
+text = st.text_input('Enter something to begin with...', placeholder='I look at her, then i realised...', key="value")
+st.write("Generating...")
 if text:
-    st.write("Generating...")
     for _ in range(50):
         result = generate(text)
         text=slice(result)
-        result = result.replace(text,"")
-        out = ""
-        for i in result:
-            if i == '.':st.title(out)
-            else: out=out+i;
+        st.title(result.replace(text,""))
         #with concurrent.futures.ThreadPoolExecutor() as executor:
         #   executor.submit(type_text, result.replace(text,""))