Spaces:

torileatherman
/

news_headline_sentiment

Runtime error

torileatherman commited on Jan 13, 2023

Commit

a3d5133

1 Parent(s): 2d76902

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,13 +8,11 @@ fs = project.get_feature_store()
 dataset_api = project.get_dataset_api()
-dataset = load_dataset("torileatherman/sentiment_analysis_batch_predictions")
-predictions_df = pd.DataFrame(dataset,columns=['Headlines_seq', 'URL','Headline_str','Predictions'])
-predictions_df_url0 = predictions_df['URL'].iloc[1]
-predictions_df_url1 = predictions_df['URL'].iloc[2]
-predictions_df_url2 = predictions_df['URL'].iloc[3]
 def article_selection(sentiment):
     if sentiment == "Positive":

 dataset_api = project.get_dataset_api()
+dataset = load_dataset("torileatherman/sentiment_analysis_batch_predictions", split='train')
+predictions_df = pd.DataFrame(dataset)
+predictions_df_url0 = predictions_df['Url'].iloc[0]
+predictions_df_url1 = predictions_df['Url'].iloc[1]
+predictions_df_url2 = predictions_df['Url'].iloc[2]
 def article_selection(sentiment):
     if sentiment == "Positive":