| from sftp import SpanPredictor | |
| predictor = SpanPredictor.from_path( | |
| # '/home/gqin2/public/release/sftp/0.0.1/framenet/model.tar.gz', # For CLSP grid | |
| "../model.mod.tar.gz", # For BRTX | |
| cuda_device=-1 # Change this to 0 for CUDA:0 | |
| ) | |
| while True: | |
| sentence = input(">>>") | |
| result = predictor.predict_sentence(sentence) | |
| for child in result.span: | |
| print(child) | |
| for sub_child in child: | |
| print("\t", sub_child) | |