Spaces:

gagan3012
/

T5-Summarization

Runtime error

Gagan Bhatia commited on Jul 23, 2021

Commit

c429702

•

1 Parent(s): 4ea49ac

Update model.py

Files changed (1) hide show

src/models/model.py CHANGED Viewed

@@ -346,6 +346,7 @@ class Summarization:
     def load_model(
             self, model_dir: str = "../../models", use_gpu: bool = False
     ):
         """
         loads a checkpoint for inferencing/prediction

     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
     ):
         """
         loads a checkpoint for inferencing/prediction