code: def __init__(self, *args, **kwargs):
      super().__init__(*args, **kwargs)
      self.model = DistilBertForQuestionAnswering.from_pretrained(
          self.model_dir
      )
      self.tokenizer = DistilBertTokenizer.from_pretrained(
          self.model_dir
      )
      self.device = torch.device(
          "cuda" if torch.cuda.is_available() else "cpu"
      )
      self.model.to(self.device)
  
file_path: nboost/nboost/plugins/qa/distilbert.py
imports: from typing import Tuple
  from transformers import DistilBertForQuestionAnswering, DistilBertTokenizer
  import numpy
  import torch
  from nboost.plugins.qa.base import QAModelPlugin
label_desc: Use image node_id 1 to create a new PtDistilBertQAModelPlugin object from inherited base classes: QAModelPlugin with example: obj = PtDistilBertQAModelPlugin()