Spaces:
Running
Running
ThanaritKanjanametawat
commited on
Commit
•
c9004d2
1
Parent(s):
445e717
Fix demo pipeline 2
Browse files
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 |
-
|
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')
|