Spaces:

gagan3012
/

T5-Summarization

Runtime error

Gagan Bhatia commited on Jul 23, 2021

Commit

ff93824

1 Parent(s): c429702

Update model.py

Files changed (1) hide show

src/models/model.py CHANGED Viewed

@@ -345,7 +345,6 @@ class Summarization:
         trainer.fit(self.T5Model, self.data_module)
     def load_model(
-            self, model_dir: str = "../../models", use_gpu: bool = False
             self, model_type: str = 't5', model_dir: str = "../../models", use_gpu: bool = False
     ):
         """

         trainer.fit(self.T5Model, self.data_module)
     def load_model(
             self, model_type: str = 't5', model_dir: str = "../../models", use_gpu: bool = False
     ):
         """