File size: 304 Bytes
fca50f9
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15

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()