Update predicator.py

predicator.py

@@ -115,7 +115,7 @@ def words_and_spans_from_tokenizer(sentence: str, nlp, tokenizer):
         
         
     doc = nlp(sentence)
-    words_final = [token for token in doc]
+    words_final = [token.text for token in doc]
     
     return words, spans, words_final
 
@@ -309,7 +309,7 @@ def print_srl_frames_pretty(words, frames, show_grid=True, color=False):
                 i += 1
         return spans
     
-    words = [word.text for word in words]
+    # words = [word.text for word in words]
     print("Sentence:", " ".join(words))
     if not frames:
         print("  (no predicates detected)")