Akshay Vs commited on
Commit
d38987e
1 Parent(s): 5c2d53d
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -5,6 +5,8 @@ import streamlit as st
5
  import concurrent.futures
6
  from random import randint
7
 
 
 
8
  if st.session_state.new==True:
9
  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')
10
  os.system('!pip install transformers')
 
5
  import concurrent.futures
6
  from random import randint
7
 
8
+ # initializing session_state
9
+ st.session_state['new']=True
10
  if st.session_state.new==True:
11
  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')
12
  os.system('!pip install transformers')