Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
•
5866066
1
Parent(s):
bcf1243
Update model.py
Browse files- src/models/model.py +1 -1
src/models/model.py
CHANGED
@@ -385,7 +385,7 @@ class Summarization:
|
|
385 |
trainer.fit(self.T5Model, self.data_module)
|
386 |
|
387 |
def load_model(
|
388 |
-
|
389 |
):
|
390 |
"""
|
391 |
loads a checkpoint for inferencing/prediction
|
|
|
385 |
trainer.fit(self.T5Model, self.data_module)
|
386 |
|
387 |
def load_model(
|
388 |
+
self, model_type: str = "t5", model_dir: str = "models", use_gpu: bool = False
|
389 |
):
|
390 |
"""
|
391 |
loads a checkpoint for inferencing/prediction
|