Update README.md

README.md

@@ -28,7 +28,7 @@ The model recipe: https://github.com/ParagEkbote/El-Emperador_ModelRecipe
 
 ## Inference Script:
 
-```yaml
+```python
 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)
@@ -56,4 +56,5 @@ if __name__ == "__main__":
     
     print(f"Input: {input_text}")
     print(f"Output: {output}")
+```