Update README.md

README.md

@@ -110,22 +110,15 @@ TRAINING Loss: 3.422200(At Step:40000)
 
 We can use the model directly with a pipeline for text generation:
 
+!pip install transformers
 >>>from transformers import GPT2Tokenizer, GPT2LMHeadModel
->>>
 >>>model_name = "Dwaraka/PROJECT_GUTENBERG_GOTHIC_FICTION_TEXT_GENERATION_gpt2"
->>>
->>>model = GPT2LMHEadModel.from_pretrained(model_name)
->>>
+>>>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 :