Spaces:
Runtime error
Runtime error
import gradio as gr | |
import emoji | |
from simpletransformers.classification import ClassificationModel, ClassificationArgs | |
CLASS_NAMES = ['angry_face', 'crying_face', 'face_with_crossed-out_eyes', 'face_with_open_mouth', 'flushed_face', 'grinning_face_with_smiling_eyes', 'loudly_crying_face', 'pouting_face', 'slightly_smiling_face', 'smiling_face_with_smiling_eyes', 'sparkles', 'tired_face'] | |
model_args = ClassificationArgs() | |
model_original = ClassificationModel( | |
'auto', | |
'xiongjie/face-expression-ja', | |
use_cuda=False, | |
args=model_args, | |
num_labels=len(CLASS_NAMES) | |
) | |
def predict_emoticon(text): | |
return emoji.emojize(":" + CLASS_NAMES[model_original.predict([text])[0][0]] + ":") | |
iface = gr.Interface(fn=predict_emoticon, inputs="text", outputs="text") | |
iface.launch() |