application_vnd.jupyter.stderr_output_3.png text_plain_output_2.png application_vnd.jupyter.stderr_output_1.png
import pandas as pd

train_df = pd.read_parquet(paths.DIFUSSION_DB_META)
train_df.rename(columns={'Prompt': 'prompt'}, inplace=True)
train_df['prompt'] = train_df['prompt'].astype(str)
train_df['prompt'] = train_df['prompt'].apply(lambda x: x.lower())
print(f'Train shape: {train_df.shape}')
train_df.head()