Spaces:
Runtime error
Runtime error
kabita-choudhary
commited on
Commit
·
74c4cd3
1
Parent(s):
20e4075
Update app.py
Browse files
app.py
CHANGED
@@ -14,7 +14,7 @@ def summarize(data, modelname):
|
|
14 |
model = PegasusForConditionalGeneration.from_pretrained("google/pegasus-xsum")
|
15 |
tokenizer = PegasusTokenizer.from_pretrained("google/pegasus-xsum")
|
16 |
else:
|
17 |
-
summarizer = pipeline("summarization", model="
|
18 |
print("world")
|
19 |
output = summarizer(data, max_length=300, min_length=30, do_sample=False)
|
20 |
return output[0]["summary_text"]
|
|
|
14 |
model = PegasusForConditionalGeneration.from_pretrained("google/pegasus-xsum")
|
15 |
tokenizer = PegasusTokenizer.from_pretrained("google/pegasus-xsum")
|
16 |
else:
|
17 |
+
summarizer = pipeline("summarization", model="lidiya/bart-large-xsum-samsum")
|
18 |
print("world")
|
19 |
output = summarizer(data, max_length=300, min_length=30, do_sample=False)
|
20 |
return output[0]["summary_text"]
|