Update README.md

README.md

@@ -70,8 +70,8 @@ device='cpu'
 model = GPT2LMHeadModel.from_pretrained('fittar/ViPE-S-CTX7')
 model.to(device)
 
-#ViPE-M's tokenizer is identical to that of GPT2-Medium
-tokenizer = GPT2Tokenizer.from_pretrained('gpt2-medium')
+#ViPE-M's tokenizer is identical to that of GPT2-Small
+tokenizer = GPT2Tokenizer.from_pretrained('gpt2')
 tokenizer.pad_token = tokenizer.eos_token
 
 # A list of abstract/figurative or any arbitrary combinations of keywords
@@ -81,6 +81,10 @@ prompts=generate(texts,model,tokenizer,do_sample=True,device=device)
 for t,p in zip(texts,prompts):
     print('{} --> {}'.format(t,p))
 
+lalala -->  A group of dancers performing an extravagant traditional dance, lalala in Spanish
+I wanna start learning -->  A student intently sitting at a desk, surrounded by books and notes
+free your mind; you will see the other side of life -->  A view of the night sky, stars and planets shining bright, while a woman in a field of flowers looks up in awe
+brave; fantasy -->  A knight in shining armor riding a gallant horse through a sunlit valley
 
 ```