Spaces:

ganeshkamath89
/

Wikipedia_Summarizer

Sleeping

ganeshkamath89 commited on Apr 24

Commit

94e1082

•

1 Parent(s): eb21b56

adding model attribute and specifying a model that is capable of taking larger texts

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ def get_wiki_summary(search):
     tittle = page.title
     text = page.text
     #summary = page.summary[0:300]
-    summarizer = pipeline("summarization")
     summary = summarizer(text)[0]["summary_text"]
     return isExist, url, tittle, summary, text

     tittle = page.title
     text = page.text
     #summary = page.summary[0:300]
+    summarizer = pipeline("summarization", model = "allenai/led-large-16384-arxiv")
     summary = summarizer(text)[0]["summary_text"]
     return isExist, url, tittle, summary, text