Update app.py

app.py

@@ -10,11 +10,11 @@ model_original = ClassificationModel(
     'xiongjie/face-expression-ja',
     use_cuda=True,
     args=model_args,
-    num_labels=len(emoticon_dataset.CLASS_NAMES)
+    num_labels=len(CLASS_NAMES)
 )
 
 def predict_emoticon(text):
-    return emoji.emojize(":" + model_onnx.predict([text])[0][0] + ":")
+    return emoji.emojize(":" + CLASS_NAMES[model_onnx.predict([text])[0][0]] + ":")
 
 iface = gr.Interface(fn=greet, inputs="text", outputs="text")
 iface.launch()