Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
·
3bd7bbf
1
Parent(s):
ff93824
Update model.py
Browse files- src/models/model.py +0 -1
src/models/model.py
CHANGED
@@ -354,7 +354,6 @@ class Summarization:
|
|
354 |
model_dir (str, optional): path to model directory. Defaults to "outputs".
|
355 |
use_gpu (bool, optional): if True, model uses gpu for inferencing/prediction. Defaults to True.
|
356 |
"""
|
357 |
-
self.model = T5ForConditionalGeneration.from_pretrained(f"{model_dir}")
|
358 |
self.tokenizer = T5Tokenizer.from_pretrained(f"{model_dir}")
|
359 |
|
360 |
if use_gpu:
|
|
|
354 |
model_dir (str, optional): path to model directory. Defaults to "outputs".
|
355 |
use_gpu (bool, optional): if True, model uses gpu for inferencing/prediction. Defaults to True.
|
356 |
"""
|
|
|
357 |
self.tokenizer = T5Tokenizer.from_pretrained(f"{model_dir}")
|
358 |
|
359 |
if use_gpu:
|