| from idiomify.fetchers import fetch_pie | |
| from preprocess import annotate | |
| def main(): | |
| pie_df = fetch_pie() | |
| pie_df = pie_df.pipe(annotate, boi_token="<idiom>", eoi_token="</idiom>") | |
| for _, row in pie_df.iterrows(): | |
| print(row['Idiomatic_Sent']) | |
| if __name__ == '__main__': | |
| main() | |