sarat2hf commited on
Commit
f2797ee
1 Parent(s): 3cc1803

Update pipeline parameters

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -11,7 +11,7 @@ from datetime import datetime, timedelta
11
  app = Flask(__name__)
12
 
13
  # Set up sentiment analysis and summarization pipelines
14
- sentiment_analysis_pipeline = pipeline("sentiment-analysis")
15
  summarization_pipeline = pipeline("summarization")
16
 
17
  # News API setup
@@ -84,11 +84,11 @@ def index():
84
  link = article['url']
85
  content = article['text']
86
 
87
- # Perform sentiment analysis on article content
88
- sentiment = sentiment_analysis_pipeline(content)[0]
89
-
90
  # Summarize article content
91
  summary = summarization_pipeline(content, max_length=100, min_length=30, do_sample=False)[0]["summary_text"]
 
 
 
92
 
93
  news_data.append({
94
  "title": title,
 
11
  app = Flask(__name__)
12
 
13
  # Set up sentiment analysis and summarization pipelines
14
+ sentiment_analysis_pipeline = pipeline("sentiment-analysis",truncation=True, max_length=512)
15
  summarization_pipeline = pipeline("summarization")
16
 
17
  # News API setup
 
84
  link = article['url']
85
  content = article['text']
86
 
 
 
 
87
  # Summarize article content
88
  summary = summarization_pipeline(content, max_length=100, min_length=30, do_sample=False)[0]["summary_text"]
89
+
90
+ # Perform sentiment analysis on article content
91
+ sentiment = sentiment_analysis_pipeline(summary)[0]
92
 
93
  news_data.append({
94
  "title": title,