Update README.md

README.md

@@ -15,5 +15,11 @@ from transformers import pipeline
 
 pipe = pipeline("text-generation", model="clfegg/gpt-tinystories-vi-subset")
 pipe("một buổi sáng, tôi",max_new_tokens=250)
-# [{'generated_text': 'một buổi sáng, tôi sẽ giúp bạn. " " Con chim nói, " " Được rồi, hãy chơi! " " Vì_vậy, con chim và con chim chơi với nhau cả ngày. Chúng rất vui_vẻ. Con chim và con chim trở_thành bạn tốt của nhau. Chúng chơi với nhau mỗi ngày. Và chúng sống hạnh_phúc mãi_mãi về sau_này.}] 
+def stories_generate(input):
+  response=pipe(input,max_new_tokens=128)[0]['generated_text']
+  text = response.split("<")[0]
+  text = text.replace("_", " ")
+  return text
+print(stories_generate("một ngày nọ"))
+# [{'generated_text': 'một ngày nọ, một cô bé tên Lily đi dạo. Cô bé nhìn thấy một cái cây lớn với rất nhiều lá. Lily muốn leo lên cây nhưng cô bé quá nhỏ. Lily nhờ mẹ giúp đỡ. Mẹ cô nói, " " Đừng lo, Lily. Chúng ta có thể trèo lên cây. " " Lily trèo lên cây và trèo lên cây. Cô trèo lên cây và trèo lên cây. Khi cô leo lên đến đỉnh, cô thấy một con chim nhỏ. Con chim nói, " " Cảm ơn, Lily! " " Lily mỉm cười và nói, " " Cảm ơn mẹ! " " Họ trở thành những người bạn tốt và chơi cùng nhau mỗi ngày. "}] 
 ```