z-dickson commited on
Commit
2a2c067
1 Parent(s): 9544a49

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -30,10 +30,10 @@ It achieves the following results on the evaluation set:
30
 
31
  ```python
32
  import torch
33
- from transformers import AutoTokenizer, AutoModelForSequenceClassification, TextClassificationPipeline,TFAutoModelForSequenceClassification, AutoTokenizer
34
 
35
  tokenizer = AutoTokenizer.from_pretrained("z-dickson/multilingual_sentiment_newspaper_headlines")
36
- m1 = TFAutoModelForSequenceClassification.from_pretrained("z-dickson/multilingual_sentiment_newspaper_headlines")
37
  sentiment_classifier = TextClassificationPipeline(tokenizer=tokenizer, model=m1)
38
 
39
  sentiment_classifier('Brazylia: Bolsonaro wci±ż nie uznał porażki. Jego zwolennicy blokuj± autostrady')
 
30
 
31
  ```python
32
  import torch
33
+ from transformers import AutoTokenizer, TextClassificationPipeline,TFAutoModelForSequenceClassification
34
 
35
  tokenizer = AutoTokenizer.from_pretrained("z-dickson/multilingual_sentiment_newspaper_headlines")
36
+ m1 = TFAutoModelForSequenceClassification.from_pretrained("z-dickson/multilingual_sentiment_newspaper_headlines", from_tf=True)
37
  sentiment_classifier = TextClassificationPipeline(tokenizer=tokenizer, model=m1)
38
 
39
  sentiment_classifier('Brazylia: Bolsonaro wci±ż nie uznał porażki. Jego zwolennicy blokuj± autostrady')