Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -92,7 +92,7 @@ def preprocess_text(text):
|
|
92 |
|
93 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
94 |
|
95 |
-
summarization_model = AutoModelForSeq2SeqLM.from_pretrained("t5-base",
|
96 |
summarization_tokenizer = AutoTokenizer.from_pretrained("t5-base")
|
97 |
summarization_model.to(device)
|
98 |
|
|
|
92 |
|
93 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
94 |
|
95 |
+
summarization_model = AutoModelForSeq2SeqLM.from_pretrained("t5-base", return_dict=True, torch_dtype=torch.float16)
|
96 |
summarization_tokenizer = AutoTokenizer.from_pretrained("t5-base")
|
97 |
summarization_model.to(device)
|
98 |
|