import gradio as gr
from transformers import pipeline


model = pipeline("text-classification", model="ProsusAI/finbert")

def input_text_fn(text):
    result = model(text)
    label = result[0]['label']
    confidence = result[0]['score']
    return f"Label: {label}, Confidence: {confidence:.2f}"


interface = gr.Interface(
    fn=input_text_fn,
    inputs="text",
    outputs="text",
    title="Financial Data Sentiment Analysis",
    description="Enter financial text to classify it as positive, neutral or negative sentiment."
)

# Launch the app
interface.launch()