Spaces:
Sleeping
Sleeping
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')
|