Update README.md

README.md

@@ -17,7 +17,8 @@ ElEmperador is an ORPO finetinue derived from the Mistral-7B-v0.1 base model.
 
 Inference Script:
 
-- def generate_response(model_name, input_text, max_new_tokens=50):
+```yaml
+def generate_response(model_name, input_text, max_new_tokens=50):
     # Load the tokenizer and model from Hugging Face Hub
     tokenizer = AutoTokenizer.from_pretrained(model_name)
     model = AutoModelForCausalLM.from_pretrained(model_name)
@@ -34,7 +35,7 @@ Inference Script:
     
     return generated_text
 
-- if __name__ == "__main__":
+if __name__ == "__main__":
     # Set the model name from Hugging Face Hub
     model_name = "AINovice2005/ElEmperador" 
     input_text = "Hello, how are you?"