Update DataLoader.py

DataLoader.py

@@ -7,7 +7,7 @@ import matplotlib.pyplot as plt
 import yfinance as yf
 from datetime import datetime
 from pygooglenews import GoogleNews
-from transformers import TFAutoTokenizer, TFAutoModelForSequenceClassification
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
 from transformers import pipeline
 class DataLoader:
     def __init__(self, ticker, time_period_news, time_period_stock, news_decay_rate = 0):
@@ -45,8 +45,8 @@ class DataLoader:
         return news_data
 
     def get_sentiment(self, news_data):
-        tokenizer = TFAutoTokenizer.from_pretrained("ProsusAI/finbert")
-        model = TFAutoModelForSequenceClassification.from_pretrained("ProsusAI/finbert")
+        tokenizer = AutoTokenizer.from_pretrained("ProsusAI/finbert")
+        model = AutoModelForSequenceClassification.from_pretrained("ProsusAI/finbert")
         classifier = pipeline('sentiment-analysis', model=model, tokenizer=tokenizer)
 
         news_sentiment = []