Update app.py

app.py

@@ -1,16 +1,18 @@
 import gradio as gr
 import hopsworks
+from datasets import load_dataset
 
 project = hopsworks.login()
 fs = project.get_feature_store()
 
 dataset_api = project.get_dataset_api()
 
-dataset_api.download("Resources/images/latest_passenger_prediction.png")
+batch_data_predictions.csv = dataset_api.download("Resources/batch_data_predictions.csv")
+dataset = load_dataset("csv", data_files="batch_data_predictions.csv")
 
 def article_selection(sentiment):
     if sentiment == "Positive":
-        return gr.image('latest_passenger_prediction.png')#f"""The sentence you requested is Positive!"""
+        return dataset #f"""The sentence you requested is Positive!"""
     #elif sentiment == "Negative":
      #   return f"""The sentence you requested is Negative!""" 
     #else:
@@ -21,7 +23,7 @@ def article_selection(sentiment):
 demo = gr.Interface(
     fn=article_selection,
     inputs = gr.Dropdown(["Positive","Negative","Neutral"], label="What type of news articles would you like recommended?"),
-    outputs = [gr.Image(label="Sentiment of News Articles")],
+    outputs = [gr.Textbox(label="Sentiment of News Articles")],
 )