Update app.py

app.py

@@ -181,7 +181,11 @@ if uploaded_file is not None:
             text = pg.get_text()
             sentences = sentence_tokenize(text)
             for sent in sentences:
+              x = mask_generation(sent)
+              t5_words = list(set(sent.split(" ")).difference(set(x.split(" "))))
+              
               entities,words_out = extract_entities(sent)
+              words_out+=t5_words
               bert_words = words_red_bert(sent)
               new=[]
               for w in words_out: