Update README.md
Browse files
README.md
CHANGED
@@ -1,5 +1,11 @@
|
|
1 |
---
|
2 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
3 |
---
|
4 |
pip install transformers
|
5 |
from transformers import automodelforcausallm, autotokenizer
|
@@ -9,5 +15,4 @@ model_name="mrm8488/distilroberta-finetuned-financial-news-sentiment-analysis"
|
|
9 |
tokenizer = autotokenizer.from_pretrained(model_name)
|
10 |
model=automodelforcausallm.from_pretrained(model_name,pad_token_id=tokenizer.eos_token_id).to(torch_device)
|
11 |
model_inputs=tokenizer('Nvidia reported profits of €10Mio',return_tensors='pt').to(torch_device)
|
12 |
-
output = model(**inputs).logits.argmax(axis=1)
|
13 |
-
|
|
|
1 |
---
|
2 |
+
language:
|
3 |
+
- en
|
4 |
+
base_model:
|
5 |
+
- distilbert/distilroberta-base
|
6 |
+
pipeline_tag: text-classification
|
7 |
+
tags:
|
8 |
+
- finance
|
9 |
---
|
10 |
pip install transformers
|
11 |
from transformers import automodelforcausallm, autotokenizer
|
|
|
15 |
tokenizer = autotokenizer.from_pretrained(model_name)
|
16 |
model=automodelforcausallm.from_pretrained(model_name,pad_token_id=tokenizer.eos_token_id).to(torch_device)
|
17 |
model_inputs=tokenizer('Nvidia reported profits of €10Mio',return_tensors='pt').to(torch_device)
|
18 |
+
output = model(**inputs).logits.argmax(axis=1)
|
|