Update app.py

app.py

@@ -1,46 +1,55 @@
-import gradio as gr
-import joblib
-
-model = joblib.load(
-"prompt_injection_multilingual.pkl"
-)
-
-def detect(prompt):
-
-```
-prob = model.predict_proba(
-    [prompt]
-)[0][1]
-
-if prob >= 0.5:
-
-    result = "🚨 Prompt Injection Detected"
-
-else:
-
-    result = "✅ Safe Prompt"
-
-return result, f"{prob:.2%}"
-```
-
-demo = gr.Interface(
-
-fn=detect,
-
-inputs="textbox",
-
-outputs=[
-
-"text",
-
-"text"
-
-],
-
-title="🛡️ Prompt Injection Detector",
-
-description="English + Urdu + Roman Urdu"
-
-)
-
-demo.launch()
+import gradio as gr
+import joblib
+
+model = joblib.load(
+"prompt_injection_multilingual.pkl"
+)
+
+def detect(prompt):
+
+```
+probability = model.predict_proba(
+    [prompt]
+)[0][1]
+
+if probability >= 0.5:
+
+    verdict = "🚨 Prompt Injection Detected"
+
+else:
+
+    verdict = "✅ Safe Prompt"
+
+return verdict, f"{probability:.2%}"
+```
+
+demo = gr.Interface(
+
+```
+fn=detect,
+
+inputs=gr.Textbox(
+    lines=5,
+    label="Enter Prompt"
+),
+
+outputs=[
+
+    gr.Textbox(
+        label="Result"
+    ),
+
+    gr.Textbox(
+        label="Injection Probability"
+    )
+
+],
+
+title="🛡️ Prompt Injection Detector",
+
+description="English + Urdu + Roman Urdu"
+```
+
+)
+
+demo.launch()