numind
/

NuSentiment

Feature Extraction

sentiment-analysis

text-classification

sentiment-classification

text-embeddings-inference

Model card Files Files and versions

Alexandre-Numind commited on Aug 11, 2023

Commit

855ca2b

·

1 Parent(s): 94a98cc

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -24,8 +24,8 @@ import torch
 from transformers import AutoTokenizer, AutoModel
-model = AutoModel.from_pretrained("Numind/e5-base-SA")
-tokenizer = AutoTokenizer.from_pretrained("Numind/e5-base-SA")
 device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
 model.to(device)

 from transformers import AutoTokenizer, AutoModel
+model = AutoModel.from_pretrained("Numind/e5-base-sentiment_analysis")
+tokenizer = AutoTokenizer.from_pretrained("Numind/e5-base-sentiment_analysis")
 device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
 model.to(device)