gradio/demo/sentiment_analysis/run.py
aliabid94 b4d9825409
Website: WIP (#328)
Ported gradio website into gradio repository, now launched as a docker service from gradio/website
2021-12-13 22:02:19 -08:00

17 lines
428 B
Python

import gradio as gr
import nltk
from nltk.sentiment.vader import SentimentIntensityAnalyzer
nltk.download('vader_lexicon')
sid = SentimentIntensityAnalyzer()
def sentiment_analysis(text):
scores = sid.polarity_scores(text)
del scores["compound"]
return scores
iface = gr.Interface(sentiment_analysis, "textbox", "label", interpretation="default")
iface.test_launch()
if __name__ == "__main__":
iface.launch()