Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ from transformers import pipeline
3
 
4
  # Summarization
5
  def summarization(text):
6
- text_model = pipeline("text-generation", model="ainize/bart-base-cnn")
7
  summary = text_model(text, max_length=100, temperature=1.0)[0]["generated_text"]
8
  return summary
9
 
 
3
 
4
  # Summarization
5
  def summarization(text):
6
+ text_model = pipeline("summarization", model="philschmid/bart-large-cnn-samsum")
7
  summary = text_model(text, max_length=100, temperature=1.0)[0]["generated_text"]
8
  return summary
9