yeomtong commited on
Commit
5a8e5d0
·
verified ·
1 Parent(s): 4c451bf

Update visualizer_dev.py

Browse files
Files changed (1) hide show
  1. visualizer_dev.py +2 -2
visualizer_dev.py CHANGED
@@ -171,7 +171,7 @@ def print_srl_frames_pretty(words, frames, show_grid=True, color=False):
171
 
172
  def prediction(model_path, bert_name, sentence):
173
  words, frames = main_predictor(model_path, bert_name, sentence)
174
- # assumes you have print_srl_frames_pretty already defined somewhere
175
  try:
176
  print_srl_frames_pretty(words, frames, show_grid=True, color=False)
177
  except NameError:
@@ -184,7 +184,7 @@ def prediction(model_path, bert_name, sentence):
184
 
185
  def prediction_formatted(model_path, bert_name, sentence):
186
  words, frames = main_predictor(model_path, bert_name, sentence)
187
- # assumes you have create_dict, otherwise return raw
188
  try:
189
  return create_dict(words, frames)
190
  except NameError:
 
171
 
172
  def prediction(model_path, bert_name, sentence):
173
  words, frames = main_predictor(model_path, bert_name, sentence)
174
+
175
  try:
176
  print_srl_frames_pretty(words, frames, show_grid=True, color=False)
177
  except NameError:
 
184
 
185
  def prediction_formatted(model_path, bert_name, sentence):
186
  words, frames = main_predictor(model_path, bert_name, sentence)
187
+
188
  try:
189
  return create_dict(words, frames)
190
  except NameError: