Spaces:
Sleeping
Sleeping
import gradio as gr | |
from transformers import pipeline | |
distilled_student_sentiment_classifier = pipeline( | |
model="lxyuan/distilbert-base-multilingual-cased-sentiments-student", | |
return_all_scores=True | |
) | |
def sentiment_analysis(user_query: str): | |
result = distilled_student_sentiment_classifier(user_query)[0] | |
structured_result = dict() | |
for r in result: | |
structured_result[r["label"]] = r["score"] | |
return structured_result | |
demo = gr.Interface( | |
fn = sentiment_analysis, | |
inputs = [gr.Textbox()], | |
outputs = [gr.Label()] | |
) | |
demo.launch() |