Test UI

app.py

@@ -6,11 +6,13 @@ import concurrent.futures
 from random import randint
 
 # initializing session_state
-st.session_state['new']=True
-if st.session_state.new==True:
-    os.system('pip install torch==1.10.2+cu113 torchvision==0.11.3+cu113 torchaudio===0.10.2+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html')
-    os.system('pip install transformers')
-    st.session_state.new=False 
+#os.system('pip install torch==1.10.2+cu113 torchvision==0.11.3+cu113 torchaudio===0.10.2+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html')
+os.system('pip install transformers')
+
+st.title("Story AI")
+st.markdown("<h1 style='text-align: center; color: white;'>I can generate intresting stories</h1>", unsafe_allow_html=True)
+st.markdown('')
+st.markdown('')
 
 from transformers import pipeline, set_seed
 
@@ -29,17 +31,14 @@ def type_text(text):
         sys.stdout.write(letter)
         time.sleep(0)
 
-text = input("Enter something to begin with... ")
-print(".\n.\n.\nGenerating\n.\n.\n.")
+#text = input("Enter something to begin with... ")
+#print(".\n.\n.\nGenerating\n.\n.\n.")
 
-for _ in range(50):
-    result = generate(text)
-    text=slice(result)
-    with concurrent.futures.ThreadPoolExecutor() as executor:
-        executor.submit(type_text, result.replace(text,""))
+#for _ in range(50):
+    #result = generate(text)
+    #text=slice(result)
+    #with concurrent.futures.ThreadPoolExecutor() as executor:
+    #    executor.submit(type_text, result.replace(text,""))
 
 #Streamlit
-st.title("Story AI")
-st.markdown("<h1 style='text-align: center; color: white;'>I can generate intresting stories</h1>", unsafe_allow_html=True)
-st.markdown('')
-st.markdown('')
+