sathwikabhavaraju2005 commited on
Commit
41694ed
·
verified ·
1 Parent(s): 7ae9ca8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -65,8 +65,8 @@ from transformers import pipeline
65
  summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
66
 
67
  def summarize_text(text):
68
- summary = summarizer(text, max_length=100, min_length=30, do_sample=False)
69
- return summary[0]['summary_text']
70
 
71
  # ------------------------------
72
  # Engagement Predictor (Mock)
 
65
  summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
66
 
67
  def summarize_text(text):
68
+ result = summarizer(text, max_length=120, min_length=40, do_sample=False)
69
+ return result[0]["summary_text"]
70
 
71
  # ------------------------------
72
  # Engagement Predictor (Mock)