Spaces:
Sleeping
Sleeping
Commit
·
03113eb
1
Parent(s):
0c4b135
Update app.py
Browse files
app.py
CHANGED
@@ -33,7 +33,7 @@ mdl = PegasusForConditionalGeneration.from_pretrained(mdl_name)
|
|
33 |
|
34 |
def summarize(text):
|
35 |
tokens = pegasus_tkn(text, truncation=True, padding="longest", return_tensors="pt")
|
36 |
-
txt_summary = mdl.generate(**tokens)
|
37 |
response = pegasus_tkn.batch_decode(txt_summary, skip_special_tokens=True)
|
38 |
return response
|
39 |
txt=grad.Textbox(lines=10, label="English", placeholder="English Text here")
|
|
|
33 |
|
34 |
def summarize(text):
|
35 |
tokens = pegasus_tkn(text, truncation=True, padding="longest", return_tensors="pt")
|
36 |
+
txt_summary = mdl.generate(**tokens, num_return_sequences=5, max_length=200, temperature=1.5,num_beams=10)
|
37 |
response = pegasus_tkn.batch_decode(txt_summary, skip_special_tokens=True)
|
38 |
return response
|
39 |
txt=grad.Textbox(lines=10, label="English", placeholder="English Text here")
|