fix bug of visualization using ddparser

zcf131016 · Sep 9, 2021 · 1ceaa81 · 1ceaa81
1 parent 714ca2c
commit 1ceaa81
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/paddlenlp/taskflow/dependency_parsing.py b/paddlenlp/taskflow/dependency_parsing.py
@@ -116,7 +116,7 @@ def __init__(self,
         rel_vocab_path = download_file(
             self._task_path, self.model + os.path.sep + "rel_vocab.json",
             URLS[self.model][0], URLS[self.model][1])
-        font_file_path = download_file(
+        self.font_file_path = download_file(
             self._task_path,
             self.model + os.path.sep + "SourceHanSansCN-Regular.ttf",
             URLS[self.model][0], URLS[self.model][1])
@@ -314,7 +314,7 @@ def _visualize(self, data):
             )
 
         self.plt = plt
-        self.font = font_manager.FontProperties(fname=font_file_path)
+        self.font = font_manager.FontProperties(fname=self.font_file_path)
         word, head, deprel = data['word'], data['head'], data['deprel']
 
         nodes = ['ROOT'] + word