p-christ commited on
Commit
48a26a2
·
1 Parent(s): fa9baae

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +2 -3
pipeline.py CHANGED
@@ -8,9 +8,8 @@ class PreTrainedPipeline():
8
  # Preload all the elements you are going to need at inference.
9
  # For instance your model, processors, tokenizer that might be needed.
10
  # This function is only called once, so do all the heavy processing I/O here"""
11
- self.model_name="p-christ/QuestionGenerationModelOnly"
12
- self.model = AutoModelForSeq2SeqLM.from_pretrained(self.model_name)
13
- self.tokenizer = AutoTokenizer.from_pretrained(self.model_name)
14
 
15
 
16
  def __call__(self, inputs: str):
 
8
  # Preload all the elements you are going to need at inference.
9
  # For instance your model, processors, tokenizer that might be needed.
10
  # This function is only called once, so do all the heavy processing I/O here"""
11
+ self.model = AutoModelForSeq2SeqLM.from_pretrained(path)
12
+ self.tokenizer = AutoTokenizer.from_pretrained(path)
 
13
 
14
 
15
  def __call__(self, inputs: str):