testing-space / app.py
spacedout-bits's picture
Add hello world AI sentiment analysis app
61eff92
Raw
History Blame Contribute Delete
1.25 kB
import gradio as gr
from transformers import pipeline
# Load a tiny, fast model — runs on CPU in seconds
sentiment = pipeline(
"sentiment-analysis",
model="distilbert-base-uncased-finetuned-sst-2-english",
)
def analyze(text: str) -> str:
if not text.strip():
return "Please enter some text."
result = sentiment(text)[0]
label = result["label"].capitalize()
score = round(result["score"] * 100, 1)
emoji = "😊" if label == "Positive" else "😞"
return f"{emoji} **{label}** ({score}% confidence)"
demo = gr.Interface(
fn=analyze,
inputs=gr.Textbox(
label="Your text",
placeholder="Type anything… e.g. 'I love building AI apps!'",
lines=3,
),
outputs=gr.Markdown(label="Sentiment"),
title="🤗 Hello World — AI Sentiment Analysis",
description=(
"Type a sentence and the model will tell you whether it sounds "
"**positive** or **negative**. Powered by DistilBERT on 🤗 Transformers."
),
examples=[
["I love learning about machine learning!"],
["This assignment is really frustrating."],
["The weather is okay today."],
],
cache_examples=False,
)
if __name__ == "__main__":
demo.launch()