camparchimedes commited on
Commit
2871675
·
verified ·
1 Parent(s): 8b6630d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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", device=device, return_dict=True, torch_dtype=torch.float16)
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