Update README.md

README.md

@@ -84,9 +84,19 @@ inputs_for_BartTokenizerFast= tokenizer(text, return_tensors="pt",padding=False,
 
 
 
+# Model
 
+from transformers import AutoModel
 
+model_name = "distilbert-base-uncased-finetuned-sst-2-english"
+model = AutoModel.from_pretrained(model_name)
+outputs = model(**inputs)
+
+print(outputs.last_hidden_state.shape)
 
+{
+  torch.Size([1, 7, 768])
+}