Update app.py

app.py

@@ -19,14 +19,12 @@ classifier = pipeline(
 
 def predict(user_input: str):
 
-    prediction = json.loads(classifier(user_input))[0]
-    label = prediction['label']
-    probability = prediction['score']
+    prediction = classifier(user_input)[0]
     
-    return f"The input is like to be a {label} with probability: {probability}"
+    return f"This is likely to be an {prediction['label']} with a probability of {prediction['score']}"
 
 
-textbox = gr.Textbox(placeholder="Enter user input presented for injection attack classification", lines=12)
+textbox = gr.Textbox(placeholder="Enter user input for injection attack classification", lines=12)
 
 interface = gr.Interface(
     inputs=textbox, fn=predict, outputs="text",