Test UI

app.py

@@ -6,13 +6,9 @@ import concurrent.futures
 from random import randint
 
 # initializing session_state
-#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 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
 
@@ -42,3 +38,10 @@ def type_text(text):
 
 #Streamlit
 
+
+st.markdown("<h1 style='text-align: center; color: white;'>Generate intresting stories with GPT</h1>", unsafe_allow_html=True)
+st.markdown('')
+st.markdown('')
+initial_text = st.text_input('Enter something to begin with...', placeholder='I look at her, then i realised', key="placeholder")
+if initial_text:
+    st.write("Generating...")