fclong commited on
Commit
db63fc6
1 Parent(s): 60959e7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
2
  from transformers import PegasusForConditionalGeneration
3
  from tokenizers_pegasus import PegasusTokenizer
4
 
5
- def summary(name):
6
  model = PegasusForConditionalGeneration.from_pretrained("IDEA-CCNL/Randeng-Pegasus-238M-Summary-Chinese")
7
  tokenizer = PegasusTokenizer.from_pretrained("IDEA-CCNL/Randeng-Pegasus-238M-Summary-Chinese")
8
  inputs = tokenizer(text, max_length=1024, return_tensors="pt")
 
2
  from transformers import PegasusForConditionalGeneration
3
  from tokenizers_pegasus import PegasusTokenizer
4
 
5
+ def summary(text):
6
  model = PegasusForConditionalGeneration.from_pretrained("IDEA-CCNL/Randeng-Pegasus-238M-Summary-Chinese")
7
  tokenizer = PegasusTokenizer.from_pretrained("IDEA-CCNL/Randeng-Pegasus-238M-Summary-Chinese")
8
  inputs = tokenizer(text, max_length=1024, return_tensors="pt")