Update README.md

README.md

@@ -14,8 +14,9 @@ license: apache-2.0
 
 Инструктивная модель на основе Vikhr-Qwen-2.5-1.5B-Instruct, обученная на русскоязычном сгенерированном датасете. Создана для ответа на запросы в роли волшебного психолога из вселенной Гарри Поттера.
 
-Пример использования:
+**Пример использования:**
 
+```py
 def generate_answer(model, tokenizer, prompt):
 
   inputs = tokenizer.apply_chat_template([{"role": "system", "content": BASE_SYSTEM_PROMPT},{"role": "user", "content": prompt}],
@@ -36,4 +37,5 @@ def generate_answer(model, tokenizer, prompt):
   with torch.no_grad():
       outputs = model.generate(**inputs, **gen_kwargs)
       outputs = outputs[:, inputs['input_ids'].shape[1]:]
-      return tokenizer.decode(outputs[0], skip_special_tokens=True)
+      return tokenizer.decode(outputs[0], skip_special_tokens=True)
+```