Update README.md

README.md

@@ -109,17 +109,18 @@ TRAINING Loss: 3.422200(At Step:40000)
 ### How to Use:
 
 We can use the model directly with a pipeline for text generation:
-
+``` python
 !pip install transformers
->>>from transformers import GPT2Tokenizer, GPT2LMHeadModel
->>>model_name = "Dwaraka/PROJECT_GUTENBERG_GOTHIC_FICTION_TEXT_GENERATION_gpt2"
->>>model = GPT2LMHeadModel.from_pretrained(model_name)
->>>tokenizer = GPT2Tokenizer.from_pretrained(model_name)
->>>prompt= "Once upon a time, in a dark and spooky castle, there lived a "
->>>input_ids = tokenizer.encode(prompt, return_tensors="pt" )
->>>output = model.generate(input_ids, max_length=50, do_sample=True)
->>>generated_text = tokenizer.decode(output[0], skip_special_tokens=True )
->>>print(generated_text)
+from transformers import GPT2Tokenizer, GPT2LMHeadModel
+model_name = "Dwaraka/PROJECT_GUTENBERG_GOTHIC_FICTION_TEXT_GENERATION_gpt2"
+model = GPT2LMHeadModel.from_pretrained(model_name)
+tokenizer = GPT2Tokenizer.from_pretrained(model_name)
+prompt= "Once upon a time, in a dark and spooky castle, there lived a "
+input_ids = tokenizer.encode(prompt, return_tensors="pt" )
+output = model.generate(input_ids, max_length=50, do_sample=True)
+generated_text = tokenizer.decode(output[0], skip_special_tokens=True )
+print(generated_text)
+```
 
 ### Github Link :
 This Fine-Tuned model is available at: https://github.com/DwarakaVelasiri/Dwaraka-PROJECT_GUTENBERG_GOTHIC_FICTION_TEXT_GENERATION_gpt2