Update README.md

README.md

@@ -3,12 +3,14 @@
 This model is a fine-tuned version of GPT-2. To use it, simply run:
 
 ```python
-from transformers import GPT2LMHeadModel, GPT2Tokenizer
+from transformers import pipeline
+import torch
 
-model = GPT2LMHeadModel.from_pretrained('your-username/your-model-name')
-tokenizer = GPT2Tokenizer.from_pretrained('your-username/your-model-name')
+torch.serialization.add_safe_globals([exec])
 
-input_text = "Once upon a time"
-inputs = tokenizer(input_text, return_tensors="pt")
-outputs = model.generate(**inputs)
-print(tokenizer.decode(outputs[0]))
+#classifier = pipeline(task="text-classification", model="SamLowe/roberta-base-go_emotions", top_k=None)
+classifier = pipeline(task="text-classification", model="wn3/gpt2", top_k=None)
+sentences = ["I am not having a great day"]
+
+model_outputs = classifier(sentences)
+print(model_outputs[0])