torileatherman's picture
Update app.py
a784a26
raw
history blame
1.01 kB
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/batch_data_predictions.csv")
dataset = load_dataset("csv", data_files="batch_data_predictions.csv")
def article_selection(sentiment):
if sentiment == "Positive":
return dataset #f"""The sentence you requested is Positive!"""
#elif sentiment == "Negative":
# return f"""The sentence you requested is Negative!"""
#else:
# return f"""The sentence you requested is Neutral!"""
demo = gr.Interface(
fn=article_selection,
inputs = gr.Dropdown(["Positive","Negative","Neutral"], label="What type of news articles would you like recommended?"),
outputs = [gr.Textbox(label="Sentiment of News Articles")],
)
#TODO
#demo = gr.TabbedInterface([url_demo, voice_demo], ["Swedish YouTube Video to English Text", "Swedish Audio to English Text"])
demo.launch()