Test UI

app.py

@@ -1,10 +1,17 @@
 import sys
 import time
+import os
 import streamlit as st
 import concurrent.futures
-from transformers import pipeline, set_seed
 from random import randint
 
+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
+
 generator = pipeline('text-generation', model='openai-gpt')
 
 def generate(initial_text, length=10, return_sequences=1):