using pipeline

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import json
-import tensorflow as tf
+# import tensorflow as tf
 # from transformers import AutoTokenizer
 # from transformers import TFAutoModelForSequenceClassification
 
@@ -36,7 +36,7 @@ input_elements = [gr.Textbox(label="Enter Username"), gr.Textbox(label="Enter Pa
 
 demo = gr.Interface(
     title="NoSQLi Detector",
-    description="DistilBERT-based NoSQL Injection Payload Detection Model ",
+    description="DistilBERT-based NoSQL Injection Payload Detection Model",
     fn=predict,
     inputs=input_elements,
     outputs=[gr.Textbox(label="Generated Payload"), gr.Textbox(label="Model Prediction"), gr.Slider(0, 100, label="Score")]