assemblyai
/

distilbert-base-uncased-sst2

Text Classification

Inference Endpoints

Model card Files Files and versions Community

Dillon Pulliam commited on Jun 14, 2021

Commit

b22ecd1

·

1 Parent(s): 26a8fce

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -5,7 +5,7 @@ This model is a fine-tuned version of [distilbert-base-uncased](https://huggingf
 To download and utilize this model for sentiment analysis please execute the following:
 ```python
 import torch.nn.functional as F
-from transformers import pipeline, AutoTokenizer, AutoModelForSequenceClassification
 tokenizer = AutoTokenizer.from_pretrained("assemblyai/distilbert-base-uncased-sst2")
 model = AutoModelForSequenceClassification.from_pretrained("assemblyai/distilbert-base-uncased-sst2")

 To download and utilize this model for sentiment analysis please execute the following:
 ```python
 import torch.nn.functional as F
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
 tokenizer = AutoTokenizer.from_pretrained("assemblyai/distilbert-base-uncased-sst2")
 model = AutoModelForSequenceClassification.from_pretrained("assemblyai/distilbert-base-uncased-sst2")