Update README.md

README.md

@@ -1,13 +1,24 @@
+---
+license: mit
+datasets:
+- Joshi-Aryan/chat_test
+language:
+- en
+library_name: transformers
+pipeline_tag: question-answering
+tags:
+- chat
+---
 #Usage
 
 <pre>
-  
-# pip install transformers accelerate
+
+pip install transformers accelerate
 from transformers import AutoTokenizer
 import transformers
 import torch
 
-model = "mlabonne/llama-2-7b-miniguanaco"
+model = "Joshi-Aryan/llama-2-7b-miniguanaco"
 prompt = "What is a large language model?"
 
 tokenizer = AutoTokenizer.from_pretrained(model)
@@ -17,7 +28,7 @@ pipeline = transformers.pipeline(
     torch_dtype=torch.float16,
     device_map="auto",
 )
-
+  
 sequences = pipeline(
     f'<s>[INST] {prompt} [/INST]',
     do_sample=True,
@@ -29,4 +40,5 @@ sequences = pipeline(
 for seq in sequences:
     print(f"Result: {seq['generated_text']}")
 
+  
 </pre>