Sa-m's picture
Update app.py
2afa359
raw
history blame contribute delete
784 Bytes
import gradio as gr
from xlm_emo.classifier import EmotionClassifier
io1 = 'huggingface/MilaNLProc/xlm-emo-t'
def emotionClassifier(text):
ec = EmotionClassifier()
translation = ec.predict([text])
res=translation[0]
return res
'''
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("MilaNLProc/xlm-emo-t")
model = AutoModelForSequenceClassification.from_pretrained("MilaNLProc/xlm-emo-t")
'''
inputs = gr.inputs.Textbox(label="Text")
outpts=gr.outputs.Textbox(label="Output")
#gr.Interface(emotionClassifier, inputs=inputs,outputs=outpts, title='Emotions Analyser',theme='peach').launch()
gr.Interface.load(io1, inputs=inputs, title="Emotions Analyser",theme='peach').launch(enable_queue=True)