Gagan Bhatia commited on
Commit
8cdbfed
1 Parent(s): 2bae531

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +1 -0
src/models/model.py CHANGED
@@ -248,6 +248,7 @@ class Summarization:
248
  """ initiates Summarization class """
249
  pass
250
 
 
251
  """
252
  loads T5/MT5 Model model for training/finetuning
253
  Args:
 
248
  """ initiates Summarization class """
249
  pass
250
 
251
+ def from_pretrained(self, model_type="t5", model_name="t5-base") -> None:
252
  """
253
  loads T5/MT5 Model model for training/finetuning
254
  Args: