Update README.md

README.md

@@ -11,7 +11,8 @@ from transformers import AutoTokenizer
 import transformers
 import torch
 model = "Zen1t/WordWeaver-AI"
-prompt = "Who is Leonardo Da Vinci?"
+prompt = "Сегодня он собирался удивить Аню сюрпризом — отвезти её в ресторан, заслуживающий особого внимания из-за своего расположения. А располагался тот на двадцать пятом этаже здания «Российской Академии Наук». Из его окон, судя по описанию, открывался изумительный вид на город, особенно ночью."
+prompt = f"<s>[INST] {prompt} [/INST]"
 tokenizer = AutoTokenizer.from_pretrained(model)
 pipeline = transformers.pipeline(
     "text-generation",
@@ -20,7 +21,7 @@ pipeline = transformers.pipeline(
     device_map="auto",
 )
 sequences = pipeline(
-    f'<s>[INST] {prompt} [/INST]',
+    prompt,
     do_sample=True,
     top_k=10,
     num_return_sequences=1,
@@ -28,14 +29,10 @@ sequences = pipeline(
     max_length=200,
 )
 for seq in sequences:
-    print(f"Result: {seq['generated_text']}")
+    print(seq['generated_text'][len(prompt):])
 ```
 
 **Output**
 ```
-Result: <s>[INST] Who is Leonardo Da Vinci? [/INST] Leonardo da Vinci (1452-1519) was a famous Italian polymath, artist, and inventor. (polymath: a person who knows a lot about many different subjects). He is considered one of the greatest painters of all time, and his paintings have had a big influence on many people who have lived after him.
-As an artist, Leonardo da Vinci painted beautiful paintings using his skills and knowledge. Some of his most famous paintings are:
-* The Mona Lisa: a painting of a woman who is smiling mysteriously, and is considered one of the greatest works of art of the Renaissance.
-* The Last Supper: a mural painting depicting the last meal Jesus shared with his twelve disciples before he died.
-* The Virgin and Child with St. Anne and St. John the
+Сегодня он собирался удивить Аню сюрпризом — отвезти ее в ресторан, заслуживающий особого внимания из-за своего расположения на двадцать пятом этаже здания «Российской Академии Наук». Из его окон, судя по описанию, открывался изумительный вид на город, особенно ночью.
 ```