Spaces:

Gladiator
/

Text-Summarizer

Runtime error

Gladiator commited on Dec 29, 2021

Commit

9471041

•

1 Parent(s): 7a7acba

fix bug

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ from extractive_summarizer.model_processors import Summarizer
 from transformers import T5Tokenizer, T5ForConditionalGeneration, T5Config
 def abstractive_summarizer(text : str, model):
     preprocess_text = text.strip().replace("\n", "")
     t5_prepared_text = "summarize: " + preprocess_text
     tokenized_text = tokenizer.encode(t5_prepared_text, return_tensors="pt").to("cpu")

 from transformers import T5Tokenizer, T5ForConditionalGeneration, T5Config
 def abstractive_summarizer(text : str, model):
+    device = torch.device("cpu")
     preprocess_text = text.strip().replace("\n", "")
     t5_prepared_text = "summarize: " + preprocess_text
     tokenized_text = tokenizer.encode(t5_prepared_text, return_tensors="pt").to("cpu")