Update visualizer_dev.py
Browse files- 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 |
-
|
| 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 |
-
|
| 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:
|