Spaces:
Sleeping
Sleeping
File size: 595 Bytes
8a28994 3aefa39 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
import gradio as gr
from transformers import pipeline
# Load our model (this will happen when the app starts)
sentiment_pipeline = pipeline("sentiment-analysis")
# Create a function that Gradio can use
def get_sentiment(input_text):
result = sentiment_pipeline(input_text)
return result[0]['label'] # Just return the label (POSITIVE or NEGATIVE)
# Launch the web UI
iface = gr.Interface(
fn=get_sentiment,
inputs="text",
outputs="text",
title="My First AI App",
description="Enter some text and I'll tell you if it's positive or negative."
)
iface.launch(share=True) |