Update app.py

app.py

@@ -17,8 +17,12 @@ classifier = pipeline(
 )
 
 def predict(user_input: str):
+
+    prediction = classifier(user_input)[0]
+    label = prediction['label']
+    probability = prediction['score']
     
-    return classifier(user_input)
+    return f"The input is like to be a {label} with probability: {probability}"
 
 
 textbox = gr.Textbox(placeholder="Enter user input presented for injection attack classification", lines=12)