Spaces:
Runtime error
Runtime error
| # 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() |