Update app.py

app.py

@@ -3,8 +3,11 @@ from transformers import pipeline
 
 pipe = pipeline("text-classification", model="p1atdev/saikyou-shield-30m")
 
-def classify(text):
-    return pipe(text)
+def classify(text): 
+    return labels = {
+        item["label"]: item["score"]
+        for item in pipe(text, top_k=2)
+    }
 
 demo = gr.Interface(
     fn=classify,