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

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +2 -2
pipeline.py CHANGED
@@ -2,8 +2,8 @@ from transformers import pipeline
2
  from utils import clean_text, generate_wordcloud
3
 
4
  # Load Hugging Face models
5
- sentiment_model = pipeline("sentiment-analysis")
6
- summarizer = pipeline("summarization")
7
 
8
  def process_comment(text):
9
  # Clean text
 
2
  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