Update app.py

app.py

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