Dimasnoufal's picture
Update app.py
72b9d14 verified
raw
history blame contribute delete
992 Bytes
# pip install transformers-interpret
from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
from transformers_interpret import SequenceClassificationExplainer
model = AutoModelForSequenceClassification.from_pretrained("indobertweet-fine-tuned")
tokenizer = AutoTokenizer.from_pretrained("indolem/indobertweet-base-uncased")
classifier = pipeline('text-classification', model=model, tokenizer=tokenizer)
def classify(text):
text = text.strip().lower()
result = classifier(text)
yhat = result[0]['label']
return result
# pip install gradio
import gradio as gr
iface = gr.Interface(
fn=classify,
inputs=[
gr.Textbox(placeholder="Lewandowski bermain buruk sekali, Xavi benar-benar marah kepadanya", label="Enter text to classify emotions", lines=5)
],
outputs=gr.Textbox(label="Classification Result"),
title="🔮 Emotion Classification",
description="Enter a text and classify its emotions."
)
iface.launch()