Update README.md

README.md

@@ -12,7 +12,7 @@ pipeline_tag: text-classification
 # Model Card for Model ID
 
 <!-- Provide a quick summary of what the model is/does. -->
-
+Transform the text into emotions
 
 
 ## Model Details
@@ -34,7 +34,11 @@ This is the model card of a 🤗 transformers model that has been pushed on the
 ### Model Sources [optional]
 
 <!-- Provide the basic links for the model. -->
-
+base_model:
+- SamLowe/roberta-base-go_emotions
+datasets:
+- mteb/tweet_sentiment_extraction
+- 
 - **Repository:** [More Information Needed]
 - **Paper [optional]:** [More Information Needed]
 - **Demo [optional]:** [More Information Needed]
@@ -46,6 +50,9 @@ This is the model card of a 🤗 transformers model that has been pushed on the
 ### Direct Use
 
 <!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
+from transformers import AutoModelForSequenceClassification
+from transformers import AutoTokenizer
+import torch
 
 [More Information Needed]
 
@@ -77,6 +84,8 @@ Users (both direct and downstream) should be made aware of the risks, biases and
 
 Use the code below to get started with the model.
 
+model2 = AutoModelForSequenceClassification.from_pretrained("Caseyishere/StoryCraft", num_labels=5)
+
 [More Information Needed]
 
 ## Training Details