ADI2005's picture
Upload 2 files
a5d1e72 verified
raw
history blame contribute delete
659 Bytes
import gradio as gr
from transformers import pipeline
# Load the sentiment analysis pipeline
sentiment_pipeline = pipeline("sentiment-analysis")
# Define the function that will use the model
def analyze_sentiment(text):
result = sentiment_pipeline(text)[0]
return {"label": result['label'], "score": result['score']}
# Create the Gradio interface
iface = gr.Interface(
fn=analyze_sentiment,
inputs=gr.Textbox(lines=2, placeholder="Enter text here..."),
outputs="json",
title="Sentiment Analysis",
description="A simple sentiment analysis web app using Hugging Face Transformers."
)
if __name__ == "__main__":
iface.launch()