Update app.py

app.py

@@ -30,7 +30,7 @@ def inference(input_text):
         logits = model(**inputs).logits
     
     predicted_class_id = logits.argmax().item()
-    output = model.config.label2id[predicted_class_id]
+    output = model.config.id2label[predicted_class_id]
     return output
 
 demo = gr.Interface(