              else:
                  batch = self._get_item(in_queue)
                  if batch == "EOF":
                      break
              batch = function(batch)
              if func_name == "process_output":
                  # Process output items to individual items from batch
                  for item in self.finalize(batch):
                      out_queue.put(item)
              else:
 
</s> remove from lib.utils import GetModel
 </s> add from lib.utils import GetModel, FaceswapError
            try:
                batch = function(batch)
            except tf_errors.UnknownError as err:
                msg = ("Tensorflow raised an unknown error. This is most likely caused by a "
                       "failure to launch cuDNN which can occur for some GPU/Tensorflow "
                       "combinations. You should enable `allow_growth` to attempt to resolve this "
                       "issue:"
                       "\nGUI: Go to Settings > Extract Plugins > Global and enable the "
                       "`allow_growth` option."
                       "\nCLI: Go to `faceswap/config/extract.ini` and change the `allow_growth "
                       "option to `True`.")
                raise FaceswapError(msg) from err
              else:
                  batch = self._get_item(in_queue)
                  if batch == "EOF":
                      break
            try:
                batch = function(batch)
            except tf_errors.UnknownError as err:
                msg = ("Tensorflow raised an unknown error. This is most likely caused by a "
                       "failure to launch cuDNN which can occur for some GPU/Tensorflow "
                       "combinations. You should enable `allow_growth` to attempt to resolve this "
                       "issue:"
                       "\nGUI: Go to Settings > Extract Plugins > Global and enable the "
                       "`allow_growth` option."
                       "\nCLI: Go to `faceswap/config/extract.ini` and change the `allow_growth "
                       "option to `True`.")
                raise FaceswapError(msg) from err
              if func_name == "process_output":
                  # Process output items to individual items from batch
                  for item in self.finalize(batch):
                      out_queue.put(item)
              else:
keep keep keep keep replace keep keep keep keep keep
 Capture cuDNN error in extract and raise a useful message 
plugins/extract/_base.py