ganeshkamath89 commited on
Commit
dea86a8
1 Parent(s): 94e1082

Changing to a different model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ def get_wiki_summary(search):
14
  tittle = page.title
15
  text = page.text
16
  #summary = page.summary[0:300]
17
- summarizer = pipeline("summarization", model = "allenai/led-large-16384-arxiv")
18
  summary = summarizer(text)[0]["summary_text"]
19
 
20
  return isExist, url, tittle, summary, text
 
14
  tittle = page.title
15
  text = page.text
16
  #summary = page.summary[0:300]
17
+ summarizer = pipeline("summarization", model = "BeenaSamuel/t5_small_cnn_multi_news_abstractive_summarizer_v2")
18
  summary = summarizer(text)[0]["summary_text"]
19
 
20
  return isExist, url, tittle, summary, text