Test UI

app.py

@@ -7,10 +7,10 @@ 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 
+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 
 
 from transformers import pipeline, set_seed
 
@@ -29,18 +29,17 @@ def type_text(text):
         sys.stdout.write(letter)
         time.sleep(0)
 
-if __name__ == "__main__":
-    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,""))
-
-    #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('')
+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,""))
+
+#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('')