region:us
# Suponiendo que tienes un modelo entrenado llamado "bert_story_generator"

contexto_inicial = "En un reino lejano, había un valiente caballero que..."
longitud_deseada = 200

# Generar historia
historia_generada = bert_story_generator.generate(contexto_inicial, max_length=longitud_deseada)

print(historia_generada)
