Update README.md

README.md

@@ -24,14 +24,14 @@ It can also be used for research and development in natural language understandi
 ## How to Use
 You can use this model directly with the transformers library as follows:
 
-## Download the model
+### Download the model
 ```python
 model_name = "Omaratef3221/flan-t5-base-dialogue-generator"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+```
 
-
-## Use with example
+### Use with example
 ```python
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 
@@ -48,4 +48,4 @@ Dialogue:
 input_ids = tokenizer(prompt, return_tensors='pt').input_ids
 output = model.generate(input_ids, top_p = 0.6, do_sample=True, temperature = 1.2, max_length = 512)
 print(tokenizer.decode(output[0], skip_special_tokens=True).replace('. ', '.\n'))
-
+```