Update app.py
Browse files
app.py
CHANGED
@@ -4,4 +4,5 @@ from transformers import pipeline
|
|
4 |
model="distilbert-base-uncased-finetuned-sst-2-english"
|
5 |
classifier = pipeline("sentiment-analysis", model=model)
|
6 |
text = st.text_input("Input a text:")
|
7 |
-
st.
|
|
|
|
4 |
model="distilbert-base-uncased-finetuned-sst-2-english"
|
5 |
classifier = pipeline("sentiment-analysis", model=model)
|
6 |
text = st.text_input("Input a text:")
|
7 |
+
if st.button("OK"):
|
8 |
+
st.write(classifier(text))
|