Spaces:

wiraindrak
/

summary-of-summarizer

Runtime error

wiraindrak commited on Oct 19, 2022

Commit

ac78664

1 Parent(s): 1abf4af

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -25,7 +25,7 @@ def summ_t5(text):
     return summary_text
 def summ_bert(text):
-    input_ids = tokenizer_bert.encode(ARTICLE_TO_SUMMARIZE, return_tensors='pt')
     summary_ids = model_bert.generate(input_ids,
                 min_length=20,
                 max_length=100,

     return summary_text
 def summ_bert(text):
+    input_ids = tokenizer_bert.encode(text, return_tensors='pt')
     summary_ids = model_bert.generate(input_ids,
                 min_length=20,
                 max_length=100,