Gagan Bhatia commited on
Commit
3bd7bbf
·
1 Parent(s): ff93824

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +0 -1
src/models/model.py CHANGED
@@ -354,7 +354,6 @@ class Summarization:
354
  model_dir (str, optional): path to model directory. Defaults to "outputs".
355
  use_gpu (bool, optional): if True, model uses gpu for inferencing/prediction. Defaults to True.
356
  """
357
- self.model = T5ForConditionalGeneration.from_pretrained(f"{model_dir}")
358
  self.tokenizer = T5Tokenizer.from_pretrained(f"{model_dir}")
359
 
360
  if use_gpu:
 
354
  model_dir (str, optional): path to model directory. Defaults to "outputs".
355
  use_gpu (bool, optional): if True, model uses gpu for inferencing/prediction. Defaults to True.
356
  """
 
357
  self.tokenizer = T5Tokenizer.from_pretrained(f"{model_dir}")
358
 
359
  if use_gpu: