import gradio as gr
from transformers import pipeline


sentiment = pipeline('sentiment-analysis')


def get_sentiment(input_text):
    return sentiment(input_text)


# Gradio 인터페이스 설정
iface = gr.Interface(fn=get_sentiment, title="Sentiment Analysis", inputs="text", outputs="text")

# 인터페이스 실행
iface.launch()