ThanaritKanjanametawat commited on
Commit
c9004d2
1 Parent(s): 445e717

Fix demo pipeline 2

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -7,9 +7,13 @@ st.title('GPT Detection Demo')
7
  # Add 4 options for 4 models
8
  option = st.sidebar.selectbox(
9
  'Which pipeline do you want to use?',
10
- ('sentiment-analysis'),
11
  )
12
 
 
 
 
 
13
 
14
  pipe = pipeline(option)
15
  text = st.text_area('Enter text here', 'Type here')
 
7
  # Add 4 options for 4 models
8
  option = st.sidebar.selectbox(
9
  'Which pipeline do you want to use?',
10
+ ('sentiment-analysis', 'ner', 'question-answering', 'text-generation'),
11
  )
12
 
13
+ option2 = st.sidebar.selectbox(
14
+ 'Which model do you want to use?',
15
+ ('gpt2', 'gpt2-medium', 'gpt2-large', 'gpt2-xl'),
16
+ )
17
 
18
  pipe = pipeline(option)
19
  text = st.text_area('Enter text here', 'Type here')