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

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -1
README.md CHANGED
@@ -29,7 +29,8 @@ It achieves the following results on the evaluation set:
29
 
30
 
31
  ```python
32
- from transformers import AutoTokenizer, AutoModelForSequenceClassification, TextClassificationPipeline
 
33
 
34
  tokenizer = AutoTokenizer.from_pretrained("z-dickson/multilingual_sentiment_newspaper_headlines")
35
  m1 = TFAutoModelForSequenceClassification.from_pretrained("z-dickson/multilingual_sentiment_newspaper_headlines")
 
29
 
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")