Update README.md

README.md

@@ -25,14 +25,15 @@ pip install transformers accelerate peft
 
 Load the model.
 ```{python}
+from transformers, import AutoTokenizer, AutoModelForCausalLM
+from peft import PeftModel, PeftConfig
+
 repo_id = "stefan-m-lenz/Qwen-2.5-7B-ICDOPS-QA-2024"
 config = PeftConfig.from_pretrained(repo_id, device_map="auto")
 model = AutoModelForCausalLM.from_pretrained(config.base_model_name_or_path, device_map="auto")
 model = PeftModel.from_pretrained(model, repo_id, device_map="auto")
 tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path, device_map="auto")
-```
 
-```{python}
 # Test input
 test_input = """Was ist der ICD-10-Code für die Tumordiagnose „Bronchialkarzinom, Hauptbronchus“? Antworte nur kurz mit dem ICD-10 Code."""