import gradio as gr
from transformers import pipeline

# 감정 분석 모델을 불러옵니다.
classifier = pipeline("sentiment-analysis")

# 예측 함수 정의
def sentiment_analysis(text):
    result = classifier(text)[0]
    label = result['label']
    score = result['score']
    return f"Label: {label}, Score: {score:.2f}"

# Gradio 인터페이스 설정
iface = gr.Interface(
    fn=sentiment_analysis,
    inputs="text",
    outputs="text",
    title="Sentiment Analysis",
    description="Enter text to analyze its sentiment.",
)

# 애플리케이션 실행
iface.launch()