# Use a pipeline as a high-level helper
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)
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. "}]