diff --git a/sloth/core/labeltool.py b/sloth/core/labeltool.py index 8017d6d..bcdfe84 100755 --- a/sloth/core/labeltool.py +++ b/sloth/core/labeltool.py @@ -297,7 +297,7 @@ class LabelTool(QObject): 'type': 'image', 'annotations': [ ], } - self._model.root_.addFile(fileitem) + self._model._root.appendFileItem(fileitem) def addVideoFile(self, fname): fileitem = { @@ -320,4 +320,4 @@ class LabelTool(QObject): fileitem['frames'].append(frame) i += 1 - self._model.root_.addFile(fileitem) + self._model._root.appendFileItem(fileitem)