Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
•
a88fd83
1
Parent(s):
4fb639f
Update model.py
Browse files- src/models/model.py +1 -0
src/models/model.py
CHANGED
@@ -269,6 +269,7 @@ class Summarization:
|
|
269 |
self.tokenizer = ByT5Tokenizer.from_pretrained(f"{model_name}")
|
270 |
self.model = T5ForConditionalGeneration.from_pretrained(
|
271 |
f"{model_name}", return_dict=True
|
|
|
272 |
|
273 |
def train(
|
274 |
self,
|
|
|
269 |
self.tokenizer = ByT5Tokenizer.from_pretrained(f"{model_name}")
|
270 |
self.model = T5ForConditionalGeneration.from_pretrained(
|
271 |
f"{model_name}", return_dict=True
|
272 |
+
)
|
273 |
|
274 |
def train(
|
275 |
self,
|