Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
•
7f493d7
1
Parent(s):
981a6e0
Update model.py
Browse files- src/models/model.py +1 -0
src/models/model.py
CHANGED
@@ -368,6 +368,7 @@ class Summarization:
|
|
368 |
self.tokenizer = ByT5Tokenizer.from_pretrained(f"{model_dir}")
|
369 |
self.model = T5ForConditionalGeneration.from_pretrained(
|
370 |
f"{model_dir}", return_dict=True
|
|
|
371 |
|
372 |
if use_gpu:
|
373 |
if torch.cuda.is_available():
|
|
|
368 |
self.tokenizer = ByT5Tokenizer.from_pretrained(f"{model_dir}")
|
369 |
self.model = T5ForConditionalGeneration.from_pretrained(
|
370 |
f"{model_dir}", return_dict=True
|
371 |
+
)
|
372 |
|
373 |
if use_gpu:
|
374 |
if torch.cuda.is_available():
|