Update README.md

README.md

@@ -1,3 +1,11 @@
+---
+license: apache-2.0
+datasets:
+- GAIR/lima
+language:
+- en
+pipeline_tag: text-generation
+---
 # Limstral 7B
 ## Mistral 7B fine-tuned on LIMA
 
@@ -21,4 +29,4 @@ gen = pipeline("text-generation", model=model, tokenizer=tokenizer, device=0)
 instruction = "Write a email to day goodbye to me boss"
 res = gen(instruction, max_new_tokens=512, temperature=0.3, top_p=0.75, top_k=40, repetition_penalty=1.2)
 print(res[0]['generated_text'])
-```
+```