search

talaa / streamlit_webpage

0 stars 0 forks source link

make a py file for Finsent for single url #3

Open talaa opened 1 year ago

talaa commented 1 year ago 
from transformers import BertTokenizer, BertForSequenceClassification
from transformers import pipeline

finbert = BertForSequenceClassification.from_pretrained('yiyanghkust/finbert-tone',num_labels=3)
tokenizer = BertTokenizer.from_pretrained('yiyanghkust/finbert-tone')

nlp = pipeline("sentiment-analysis", model=finbert, tokenizer=tokenizer)``