File size: 665 Bytes
2072bc5 e927444 2072bc5 e927444 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
---
language:
- en
base_model:
- distilbert/distilroberta-base
pipeline_tag: text-classification
tags:
- finance
---
pip install transformers
from transformers import automodelforcausallm, autotokenizer
import torch
torch_device = "cuda" if torch.cuda.is_available() else "cpu"
model_name="mrm8488/distilroberta-finetuned-financial-news-sentiment-analysis"
tokenizer = autotokenizer.from_pretrained(model_name)
model=automodelforcausallm.from_pretrained(model_name,pad_token_id=tokenizer.eos_token_id).to(torch_device)
model_inputs=tokenizer('Nvidia reported profits of €10Mio',return_tensors='pt').to(torch_device)
output = model(**inputs).logits.argmax(axis=1) |