Update chemietoolkit/interface.py

chemietoolkit/interface.py

@@ -7,7 +7,6 @@ from PIL import Image
 from huggingface_hub import hf_hub_download, snapshot_download
 from molscribe import MolScribe
 from rxnscribe import RxnScribe, MolDetect
-from chemiener import ChemNER
 from .tableextractor import TableExtractor
 from .utils import *
 
@@ -22,7 +21,6 @@ class ChemIEToolkit:
         self._rxnscribe = None
         self._pdfparser = None
         self._moldet = None
-        self._chemner = None
         self._coref = None
 
     @property
@@ -115,26 +113,6 @@ class ChemIEToolkit:
 
 
 
-
-
-
-    @property
-    def chemner(self):
-        if self._chemner is None:
-            self.init_chemner()
-        return self._chemner
-
-    @lru_cache(maxsize=None)
-    def init_chemner(self, ckpt_path=None):
-        """
-        Set model to custom checkpoint
-        Parameters:
-            ckpt_path: path to checkpoint to use, if None then will use default
-        """
-        if ckpt_path is None:
-            ckpt_path = hf_hub_download("Ozymandias314/ChemNERckpt", "best.ckpt")
-        self._chemner = ChemNER(ckpt_path, device=self.device)
-
     
     @property
     def tableextractor(self):