jbraha commited on
Commit
93f97ce
1 Parent(s): 6fd875d

roberta and torch

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +2 -1
app.py CHANGED
@@ -23,7 +23,7 @@ input = st.text_area("insert text to be analyzed", value="Nice to see you today.
23
 
24
  option = st.selectbox(
25
  'Choose a transformer model:',
26
- ('Default', 'Fine-Tuned' , 'Custom'))
27
 
28
 
29
  if option == 'Fine-Tuned':
 
23
 
24
  option = st.selectbox(
25
  'Choose a transformer model:',
26
+ ('Default', 'Fine-Tuned' , 'Roberta'))
27
 
28
 
29
  if option == 'Fine-Tuned':
requirements.txt CHANGED
@@ -1,3 +1,4 @@
1
  streamlit
2
  transformers
3
- tensorflow
 
 
1
  streamlit
2
  transformers
3
+ tensorflow
4
+ torch