|
|
|
|
|
|
|
|
|
|
|
from transformers import pipeline |
|
|
import torch |
|
|
import gradio as gr |
|
|
|
|
|
|
|
|
sentiment_Analyzer = pipeline("text-classification", model="distilbert/distilbert-base-uncased-finetuned-sst-2-english") |
|
|
|
|
|
|
|
|
|
|
|
def analyzer(text): |
|
|
output = (sentiment_Analyzer(text))[0] |
|
|
label = output['label'] |
|
|
score = output['score'] |
|
|
return label,score |
|
|
|
|
|
|
|
|
interface = gr.Interface( |
|
|
fn=analyzer, |
|
|
inputs=gr.Textbox(lines=2, placeholder="Enter text here..."), |
|
|
outputs=[ |
|
|
gr.Textbox(label="Sentiment Label", placeholder="Label will appear here..."), |
|
|
gr.Number(label="Confidence Score") |
|
|
], |
|
|
title="Sentiment Analyzer", |
|
|
description="Enter text to analyze its sentiment (positive/negative) and get the confidence score." |
|
|
) |
|
|
|
|
|
|
|
|
interface.launch() |