Pro-Coder commited on
Commit
346bce9
·
verified ·
1 Parent(s): feb5449

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -3,7 +3,7 @@ from utils import clean_text, generate_wordcloud
3
 
4
  # Load Hugging Face models
5
  sentiment_model = pipeline("sentiment-analysis", model="distilbert-base-uncased-finetuned-sst-2-english")
6
- summarizer = pipeline("summarization", model="facebook/bart-base")
7
 
8
  def process_comment(text):
9
  # Clean text
 
3
 
4
  # Load Hugging Face models
5
  sentiment_model = pipeline("sentiment-analysis", model="distilbert-base-uncased-finetuned-sst-2-english")
6
+ summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
7
 
8
  def process_comment(text):
9
  # Clean text