Update README.md

README.md

@@ -44,8 +44,10 @@ model = PeftModel.from_pretrained(base_model, peft_model_name)
 
 
 import torch
+
 def chat_with_model(prompt, max_length=512):
     inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
+    
     with torch.no_grad():
         output = model.generate(
             inputs['input_ids'],
@@ -60,6 +62,7 @@ def chat_with_model(prompt, max_length=512):
         )
     response = tokenizer.decode(output[0], skip_special_tokens=True)
     return response
+    
 user_input = "Hello! How are you?"
 response = chat_with_model(user_input)
 print(response)