Update app.py

app.py

@@ -2,6 +2,8 @@ import gradio as gr
 from datasets import load_dataset
 import pandas as pd
 from huggingface_hub import create_repo
+from huggingface_hub import login
+login()
 
 dataset = load_dataset("torileatherman/sentiment_analysis_batch_predictions", split='train')
 predictions_df = pd.DataFrame(dataset)
@@ -15,7 +17,7 @@ predictions_df['Prediction'] = predictions_df['Prediction'].map({0: 'Negative',
 
 def article_selection(sentiment):
     if sentiment == "Positive":
-        predictions = negative_preds
+        predictions = positive_preds
         top3 = predictions[0:3]
         top3_result = top3[['Headline_string','Url']]
         top3_result.rename(columns = {'Headline_string':'Headlines', 'Url':'URL'})
@@ -28,7 +30,7 @@ def article_selection(sentiment):
         top3_result.rename(columns = {'Headline_string':'Headlines', 'Url':'URL'})
         return top3_result
     else:
-        predictions = negative_preds
+        predictions = neutral_preds
         top3 = predictions[0:3]
         top3_result = top3[['Headline_string','Url']]
         top3_result.rename(columns = {'Headline_string':'Headlines', 'Url':'URL'})