Gagan Bhatia commited on
Commit
5ecdcfb
1 Parent(s): 26d8504

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +1 -0
src/models/model.py CHANGED
@@ -259,6 +259,7 @@ class Summarization:
259
  self.tokenizer = T5Tokenizer.from_pretrained(f"{model_name}")
260
  self.model = T5ForConditionalGeneration.from_pretrained(
261
  f"{model_name}", return_dict=True
 
262
 
263
  def train(
264
  self,
 
259
  self.tokenizer = T5Tokenizer.from_pretrained(f"{model_name}")
260
  self.model = T5ForConditionalGeneration.from_pretrained(
261
  f"{model_name}", return_dict=True
262
+ )
263
 
264
  def train(
265
  self,