Added functionality

app.py

@@ -9,7 +9,7 @@ st.markdown("<h1 style='text-align: center; color: white;'>Generate intresting s
 st.markdown('')
 st.markdown('')
 
-#initializing session_state
+# initializing session_state
 os.system('pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cpu')
 os.system('pip install transformers')
 
@@ -17,7 +17,7 @@ from transformers import pipeline, set_seed
 
 generator = pipeline('text-generation', model='openai-gpt')
 
-def generate(initial_text, length=50, return_sequences=1):
+def generate(initial_text, length=10, 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,11 +33,14 @@ def type_text(text):
 #text = input("Enter something to begin with... ")
 #print(".\n.\n.\nGenerating\n.\n.\n.")
 
-for _ in range(10):
-    result = generate(text)
-    text=slice(result)
-    st.title(result)
-    #with concurrent.futures.ThreadPoolExecutor() as executor:
-    #    executor.submit(type_text, result.replace(text,""))
+text = st.text_input('Enter something to begin with...', placeholder='I look at her, then i realised...', key="value")
+st.write("Generating...")
+if text:
+    for _ in range(50):
+        result = generate(text)
+        text=slice(result)
+        st.title(result.replace(text,""))
+        #with concurrent.futures.ThreadPoolExecutor() as executor:
+        #   executor.submit(type_text, result.replace(text,""))