Update README.md

README.md

@@ -29,10 +29,9 @@ More information needed
 
 # How to run in Google Colab
 
+```python
 from transformers import AutoTokenizer, AutoModelForCausalLM
-
 tokenizer_fromhub = AutoTokenizer.from_pretrained("suarkadipa/GPT-2-finetuned-papers")
-
 model_fromhub = AutoModelForCausalLM.from_pretrained("suarkadipa/GPT-2-finetuned-papers", from_tf=True)
 
 text_generator = pipeline(
@@ -43,11 +42,9 @@ text_generator = pipeline(
     max_new_tokens=3000
 )
 
-// change with your text
-
 test_sentence = "the lungs"
-
 res=text_generator(test_sentence)[0]["generated_text"].replace("\n", " ")
+```
 
 ## Training and evaluation data