api-model / app.py
niejanee's picture
Update app.py
0b3c2ba verified
raw
history blame contribute delete
652 Bytes
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()