unemonti commited on
Commit
e927444
·
verified ·
1 Parent(s): 2072bc5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -3
README.md CHANGED
@@ -1,5 +1,11 @@
1
  ---
2
- license: unknown
 
 
 
 
 
 
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)