generate_data.py

import pandas

df = pandas.read_parquet("data.parquet")

f = open("train_data.txt", "w", encoding="utf-8")

count = 0

for row in df.iterrows():
    f.write(row[1]["text"] + "\n")
    count += 1
    if count == 25: break

f.close()