Spaces:

wiraindrak
/

summary-of-summarizer

Runtime error

wiraindrak commited on Oct 19, 2022

Commit

6fa9846

1 Parent(s): 74899d4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -29,7 +29,7 @@ def summ_t5(text):
     return summary_text
 def summ_bert(text):
-    encoding = tokenizer_bert(text, return_tensors="pt")
     summary_ids= model_bert.generate(input_ids,
                 min_length=20,
                 max_length=100,
@@ -48,7 +48,7 @@ def summ_bert(text):
     return summary_text
 def para_t5(text):
-    input_ids = t5_para_tokenizer.encode(text, padding='longest', return_tensors='pt')
     outputs = t5_para_model.generate(
                 input_ids=encoding["input_ids"], attention_mask=encoding["attention_mask"],
                 max_length=512,

     return summary_text
 def summ_bert(text):
+    input_ids = tokenizer_bert(text, return_tensors="pt")
     summary_ids= model_bert.generate(input_ids,
                 min_length=20,
                 max_length=100,
     return summary_text
 def para_t5(text):
+    encoding = t5_para_tokenizer.encode(text, padding='longest', return_tensors='pt')
     outputs = t5_para_model.generate(
                 input_ids=encoding["input_ids"], attention_mask=encoding["attention_mask"],
                 max_length=512,