gradio/demo/sentiment_analysis.py

19 lines
458 B
Python
Raw Normal View History

2020-10-22 20:07:43 +08:00
# Demo: (Textbox) -> (Label)
2020-09-16 01:37:02 +08:00
import gradio as gr
import nltk
from nltk.sentiment.vader import SentimentIntensityAnalyzer
nltk.download('vader_lexicon')
sid = SentimentIntensityAnalyzer()
def sentiment_analysis(text):
2020-09-17 07:43:37 +08:00
scores = sid.polarity_scores(text)
del scores["compound"]
return scores
2020-09-16 01:37:02 +08:00
2020-11-11 22:15:53 +08:00
iface = gr.Interface(sentiment_analysis, "textbox", "label", interpretation="default")
2020-09-16 01:37:02 +08:00
2020-11-11 22:15:53 +08:00
iface.test_launch()
if __name__ == "__main__":
iface.launch()