Spaces:

Gladiator
/

Text-Summarizer

Runtime error

Gladiator commited on Dec 29, 2021

Commit

7a7acba

1 Parent(s): e36f01a

fix bug

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ 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(device)
     # summmarize
     summary_ids = model.generate(tokenized_text,

     preprocess_text = text.strip().replace("\n", "")
     t5_prepared_text = "summarize: " + preprocess_text
+    tokenized_text = tokenizer.encode(t5_prepared_text, return_tensors="pt").to("cpu")
     # summmarize
     summary_ids = model.generate(tokenized_text,