shubhendu-ghosh-DS commited on
Commit
e5924b8
1 Parent(s): 28e4126

change model

Browse files
summerizer_service/summarize_text.py CHANGED
@@ -8,6 +8,6 @@ class Summarizer:
8
  pass
9
 
10
  def summarize_text(self, text):
11
- summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
12
  summary = summarizer(text, max_length=500, min_length=100, do_sample=False)
13
  return summary[0]['summary_text']
 
8
  pass
9
 
10
  def summarize_text(self, text):
11
+ summarizer = pipeline("summarization", model="Falconsai/text_summarization")
12
  summary = summarizer(text, max_length=500, min_length=100, do_sample=False)
13
  return summary[0]['summary_text']