PKaushik commited on
Commit
ba0175e
·
1 Parent(s): c2946db

Create new file

Browse files
Files changed (1) hide show
  1. app.py +9 -0
app.py ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ from transformers import BertTokenizer, BertForSequenceClassification, pipeline
3
+ finbert = BertForSequenceClassification.from_pretrained('yiyanghkust/finbert-esg',num_labels=4)
4
+ tokenizer = BertTokenizer.from_pretrained('yiyanghkust/finbert-esg')
5
+ nlp = pipeline("text-classification", model=finbert, tokenizer=tokenizer)
6
+ text = st.text_area('enter some text:')
7
+ if text:
8
+ results = nlp([text])
9
+ st.json(out)