import gradio as gr
from transformers import pipeline

# Muat pipeline langsung dari model Anda di Hub
# Spaces akan otomatis men-download model Anda
pipe = pipeline("text-classification", model="niejanee/indobert-sentiment-tokopedia-reviews")

def classify_review(text):
    result = pipe(text)[0]
    return {result['label']: result['score']}

iface = gr.Interface(
    fn=classify_review,
    inputs=gr.Textbox(lines=5, placeholder="Masukkan ulasan produk di sini..."),
    outputs="label",
    title="Analisis Sentimen Ulasan Produk",
    description="Model ini menganalisis sentimen ulasan produk menggunakan fine-tuned IndoBERT."
)
iface.launch()