Update app.py
Browse files
app.py
CHANGED
@@ -1,8 +1,6 @@
|
|
1 |
-
from transformers import
|
2 |
-
|
3 |
-
tokenizer = BertTokenizer.from_pretrained('yiyanghkust/finbert-esg')
|
4 |
-
nlp = pipeline("text-classification", model=finbert, tokenizer=tokenizer)
|
5 |
text = st.text_area('enter some text:')
|
6 |
if text:
|
7 |
-
results =
|
8 |
st.json(out)
|
|
|
1 |
+
from transformers import pipeline
|
2 |
+
classifier = pipeline("sentiment-analysis", model="distilbert-base-uncased-finetuned-sst-2-english")
|
|
|
|
|
3 |
text = st.text_area('enter some text:')
|
4 |
if text:
|
5 |
+
results = classifier([text])
|
6 |
st.json(out)
|