|
from transformers import pipeline |
|
import gradio as gr |
|
|
|
model_checkpoint = "MuntasirHossain/RoBERTa-base-finetuned-emotion" |
|
|
|
model = pipeline("text-classification", model=model_checkpoint) |
|
|
|
|
|
def classify(text): |
|
label = model(text)[0]["label"] |
|
return label |
|
|
|
description = "This AI model is trained to classify texts expressing human emotion into six categories: sadness, joy, love, anger, fear, and surprise." |
|
title = "Classify Texts Expressing Emotion" |
|
|
|
examples=[["This is such a beautiful place"]] |
|
|
|
gr.Interface(fn=classify, |
|
inputs="textbox", |
|
outputs="text", |
|
title=title, |
|
|
|
description=description, |
|
examples=examples, |
|
).launch() |
