Update README.md

README.md

@@ -74,10 +74,13 @@ Collaiborator-MEDLLM-Llama-3-8b-v1 was trained using an NVIDIA A40 GPU, which pr
 ## How to use
 
 import transformers
+
 import torch
 
+
 model_id = "collaiborateorg/Collaiborator-MEDLLM-Llama-3-8B-v1"
 
+
 pipeline = transformers.pipeline(
     "text-generation",
     model=model_id,
@@ -85,22 +88,26 @@ pipeline = transformers.pipeline(
     device_map="auto",
 )
 
+
 messages = [
     {"role": "system", "content": "You are an expert trained on healthcare and biomedical domain!"},
     {"role": "user", "content": "I'm a 35-year-old male and for the past few months, I've been experiencing fatigue, increased sensitivity to cold, and dry, itchy skin. What is the diagnosis here?"},
 ]
 
+
 prompt = pipeline.tokenizer.apply_chat_template(
         messages, 
         tokenize=False, 
         add_generation_prompt=True
 )
 
+
 terminators = [
     pipeline.tokenizer.eos_token_id,
     pipeline.tokenizer.convert_tokens_to_ids("<|eot_id|>")
 ]
 
+
 outputs = pipeline(
     prompt,
     max_new_tokens=256,
@@ -109,6 +116,7 @@ outputs = pipeline(
     temperature=0.6,
     top_p=0.9,
 )
+
 print(outputs[0]["generated_text"][len(prompt):])
 
 ### Contact Information