Update README.md

README.md

@@ -99,10 +99,17 @@ print(outputs.last_hidden_state.shape)
 }
 
 
+from transformers import AutoModelForSequenceClassification
 
+model_name = "distilbert-base-uncased-finetuned-sst-2-english"
+model = AutoModelForSequenceClassification.from_pretrained(model_name)
+outputs = model(**inputs)
 
+print(outputs.logits)
 
-
+{
+  tensor([[-4.3450,  4.6878]], grad_fn=<AddmmBackward0>)
+}
 
 
 ```