from tqdm import tqdm_notebook as tqdm


remade_sentences = []
sentences = []
head_sentences = []
with open(r"C:\Users\Paul\Desktop\seq2label\cs230-code-examples\pytorch\nlp\data\small\test\sentences.txt", "r") as f:
    sentences = f.read().splitlines()
print(len(sentences[3].split()))
for j in tqdm(range(len(sentences))):
    sentence = cube(sentences[j])
    head_sentences.append([])
    for x in sentence:
        new_sent = []
        for i, word in enumerate(x):
            new_sent.append(x[i].word)
            head_sentences[j].append(x[i].head)
        remade_sentences.append(' '.join(new_sent))