Update README.md

README.md

@@ -3,6 +3,7 @@ license: mit
 language:
 - en
 base_model: meta-llama/Meta-Llama-3.1-8B-Instruct
+library_name: transformers
 ---
 
 # LLaMA 3 8B - ChatDoctor Model
@@ -33,9 +34,4 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 
 # Load the fine-tuned model and tokenizer
 model = AutoModelForCausalLM.from_pretrained("abhiyanta/llama-chatdoctor")
-tokenizer = AutoTokenizer.from_pretrained("abhiyanta/llama-chatdoctor")
-
-# Generate text
-inputs = tokenizer("What are the symptoms of diabetes?", return_tensors="pt")
-outputs = model.generate(**inputs, max_new_tokens=50)
-print(tokenizer.decode(outputs[0]))
+tokenizer = AutoTokenizer.from_pretrained("abhiyanta/llama-chatdoctor")