Update app.py

app.py

@@ -40,11 +40,7 @@ def inference(prompt):
     predicted_class_id = bi_prob.argmax().item()
     class_id = model.config.id2label[predicted_class_id]
 
-    return {
-        "class_id": class_id,
-        "clean_prob": bi_prob[0][0].item(),
-        "unclean_prob": bi_prob[0][1].item()
-        }
+    return "class_id: " + str(class_id) + "\n" + "clean_prob: " + str(bi_prob[0][0].item()) + "\n" + "unclean_prob: " + str(bi_prob[0][1].item())
 
 demo = gr.Interface(
     fn=inference,