Update README.md

README.md

@@ -56,27 +56,21 @@ How to Get Started with the Model
 
 To get started with the MyFriend model, use the code snippet below
 
+    import torch
+    from transformers import pipeline
 
+    pipe = pipeline("text-generation", model="umarmajeedofficial/MyFriend", torch_dtype=torch.bfloat16, device_map="auto")
 
-
-import torch
-from transformers import pipeline
-
-pipe = pipeline("text-generation", model="umarmajeedofficial/MyFriend", torch_dtype=torch.bfloat16, device_map="auto")
-
-messages = [
+    messages = [
     {
         "role": "system",
         "content": "You are an emergency response assistant with expertise in environmental issues.",
     },
     {"role": "user", "content": "What should I do during a heat wave?"},
-]
-prompt = pipe.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
-outputs = pipe(prompt, max_new_tokens=256, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
-print(outputs[0]["generated_text"])
-
-
-
+    ]
+    prompt = pipe.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
+    outputs = pipe(prompt, max_new_tokens=256, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
+    print(outputs[0]["generated_text"])