Donlapark commited on
Commit
4d51e1a
·
1 Parent(s): eedb662

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -1,8 +1,6 @@
1
- from transformers import BertTokenizer, BertForSequenceClassification, pipeline
2
- finbert = BertForSequenceClassification.from_pretrained('yiyanghkust/finbert-esg',num_labels=4)
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 = nlp([text])
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)