Update README.md

README.md

@@ -22,6 +22,9 @@ This model is a fine-tuned version of **`meta-llama/Llama-3.2-1B`** optimized fo
 ### Direct Use
 
 ```python
+from transformers import pipeline
+
+
 model_id_new = "Vedant3907/Llama-3.2-1B-PersonaClassifier"
 
 tokenzier = AutoTokenizer.from_pretrained(model_id_new)
@@ -44,7 +47,7 @@ pipe = pipeline(task="text-generation",
                         temperature=0.1,
                         pad_token_id = tokenizer.eos_token_id)
 
-result = pipe(testing_prompt)
+result = pipe(prompt)
 
 print(extract_labels(result[0]['generated_text']))