diff --git a/predict_links.py b/predict_links.py index 8f637a4bc8bc3cb5952c8d1b3e1b0e7cfc00bfff..4e87a1f5fae1892530e929dbf778e648531fa8dc 100644 --- a/predict_links.py +++ b/predict_links.py @@ -22,4 +22,4 @@ if __name__== '__main__': categories, links = linker.predict_without_categories(a_s) #links = linker.predict_with_categories(a_s, tags_s) idx=0 - draw_sentence_output(a_s[idx].split(" "), categories[idx], links[:,idx,:].numpy()) \ No newline at end of file + draw_sentence_output(a_s[idx].split(" "), categories[idx], links[:,idx,:].numpy())