Models modification

app.py

@@ -7,12 +7,19 @@ from newspaper import Article
 from newspaper import Config
 from bs4 import BeautifulSoup
 from datetime import datetime, timedelta
+from transformers import BertTokenizer, BertForSequenceClassification
+from transformers import BartForConditionalGeneration, BartTokenizer
+
 
 app = Flask(__name__)
 
 # Set up sentiment analysis and summarization pipelines
-sentiment_analysis_pipeline = pipeline("sentiment-analysis",truncation=True, max_length=512)
-summarization_pipeline = pipeline("summarization")
+finbert = BertForSequenceClassification.from_pretrained('yiyanghkust/finbert-tone',num_labels=3)
+sentiment_tokenizer = BertTokenizer.from_pretrained('yiyanghkust/finbert-tone')
+summarization_model = BartForConditionalGeneration.from_pretrained("facebook/bart-large", forced_bos_token_id=0)
+summarization_tokenizer = BartTokenizer.from_pretrained("facebook/bart-large")
+sentiment_analysis_pipeline = pipeline("sentiment-analysis",model=finbert,tokenizer=sentiment_tokenizer,truncation = True,max_length = 512)
+summarization_pipeline = pipeline("summarization",model=summarization_model,tokenizer=summarization_tokenizer,max_length = 512,truncation=True)
 
 # News API setup
 def get_news_articles_info(ticker_name):
@@ -84,18 +91,19 @@ def index():
  link = article['url']
  content = article['text']
  
+ # Perform sentiment analysis on article content
+ sentiment = sentiment_analysis_pipeline(content)[0]
+ 
  # Summarize article content
  summary = summarization_pipeline(content, max_length=100, min_length=30, do_sample=False)[0]["summary_text"]
-
- # Perform sentiment analysis on article content
- sentiment = sentiment_analysis_pipeline(summary)[0]
  
  news_data.append({
  "title": title,
  "link": link,
  "sentiment": sentiment["label"],
  "sentiment_score": round(sentiment["score"],3),
- "summary": summary
+ "summary": summary,
+ "Ticker": ticker
  })
 
  print(link)
@@ -124,7 +132,7 @@ def index():
  # convert the fig to HTML DIV element
  graph_html = fig.to_html(full_html=False)
  
- return render_template("index.html", news_data=news_data, candlestick_graph=graph_html)
+ return render_template("index.html", news_data=news_data, candlestick_graph=graph_html,ticker = ticker)
  
  return render_template("index.html")