Update README.md

README.md

@@ -1,6 +1,10 @@
 ---
 base_model: mistralai/Mistral-7B-Instruct-v0.1
 library_name: peft
+datasets:
+- sweatSmile/sarcastic-dataset
+language:
+- en
 ---
 
 # Model Card for Mistral-7B-Instruct-v0.1-Sarcasm
@@ -78,4 +82,4 @@ tokenizer = AutoTokenizer.from_pretrained("sweatSmile/Mistral-7B-Instruct-v0.1-S
 prompt = "Oh sure, waking up at 6am on a weekend sounds like a dream come true."
 inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
 outputs = model.generate(**inputs, max_new_tokens=50)
-print(tokenizer.decode(outputs[0], skip_special_tokens=True))
+print(tokenizer.decode(outputs[0], skip_special_tokens=True))