Spaces:
Sleeping
Sleeping
from transformers import pipeline | |
import gradio as gr | |
# Load the model from Hugging Face | |
classifier = pipeline("sentiment-analysis") | |
# Define the function to make predictions | |
def predict(text): | |
result = classifier(text)[0] | |
return f"{result['label']} ({result['score']:.2f})" | |
# Create the Gradio interface | |
gr.Interface( | |
fn=predict, | |
inputs=gr.Textbox(label="Enter Text"), | |
outputs=gr.Label(label="Sentiment"), | |
title="Sentiment Analysis with DistilBERT", | |
description="Classify text as positive or negative.", | |
).launch() |