Add pipeline tag and library name (#1)

- Add pipeline tag and library name (4c75624ee4274fa8463895bd117242f5cbf89a8e)


Co-authored-by: Niels Rogge <[email protected]>

README.md

@@ -1,8 +1,10 @@
 ---
-license: apache-2.0
 datasets:
 - HuggingFaceFW/fineweb-edu
 - yahma/alpaca-cleaned
+license: apache-2.0
+pipeline_tag: text-generation
+library_name: transformers
 ---
 
 # DMaS-LLaMa-Lite-step-43.5k-instruct
@@ -53,7 +55,10 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
 
 # Define the prompt in Vicuna 1.1 format
-prompt = "A chat between a curious user and an artificial intelligence assistant. The assistant gives helpful, detailed, and polite answers to the user's questions.\n\nUSER: What are the Pyramids of Giza known for?\nASSISTANT:"
+prompt = "A chat between a curious user and an artificial intelligence assistant. The assistant gives helpful, detailed, and polite answers to the user's questions.
+
+USER: What are the Pyramids of Giza known for?
+ASSISTANT:"
 inputs = tokenizer(prompt, return_tensors="pt")
 outputs = model.generate(**inputs, max_length=100)