File size: 427 Bytes
0a55553 feb9889 0a55553 feb9889 |
1 2 3 4 5 6 7 8 9 10 11 |
import gradio as gr
def predict_emotion(text):
inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True, max_length=128)
outputs = model(**inputs)
prediction = outputs.logits.argmax(-1).item()
return dataset["train"].features["label"].int2str(prediction)
interface = gr.Interface(fn=predict_emotion, inputs="text", outputs="label", title="Emotion Classifier")
interface.launch(share=False)
|