Update app.py

app.py

@@ -9,6 +9,7 @@ def classify(input):
     labels,values = model.predict(input, k=5)
     paired = zip(labels, values)
     label_dict = dict(paired)
+    label_dict = {k.removeprefix('__label__'): v for k, v in label_dict.items()}
     return label_dict
 
 iface = gr.Interface(fn=classify, inputs="text", outputs="label")