Update app.py

app.py

@@ -6,7 +6,7 @@ pipeline = pipeline(task="text-classification", model="TostAI/nsfw-text-detectio
 
 def predict(input_txt):
     predictions = pipeline(input_txt)
-    return input_txt, {p["label"]: p["score"] for p in predictions} 
+    return {p["label"]: p["score"] for p in predictions} 
 
 gradio_app = gr.Interface(
     predict,