Update app.py

app.py

@@ -21,7 +21,7 @@ def predict(user_input: str):
 
     prediction = classifier(user_input)[0]
     
-    return f"This is likely to be an {prediction['label']} with a probability of {prediction['score']}"
+    return f"Label: {prediction['label']}\n Probability: {round(prediction['score'], 3)}"
 
 
 textbox = gr.Textbox(placeholder="Enter user input for injection attack classification", lines=12)