Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
•
8cdbfed
1
Parent(s):
2bae531
Update model.py
Browse files- 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:
|