Update app.py

app.py

@@ -4,18 +4,18 @@ from transformers import pipeline
 
 pipe = pipeline(task="text-classification", model="TostAI/nsfw-text-detection-large")
 
-#def predict(input_txt):
-#    predictions = pipe(input_txt)
-#    return {p["label"]: p["score"] for p in predictions} 
+def predict(input_txt):
+    predictions = pipe(input_txt)
+    return {p["label"]: p["score"] for p in predictions[0]} 
 
-#gradio_app = gr.Interface(
-#    predict,
-#    inputs=gr.Textbox(label="Input text"),
-#    outputs=gr.Label(label="Result"),
-#    title="NSFW Prediction",
-#)
+gradio_app = gr.Interface(
+    predict,
+    inputs=gr.Textbox(label="Input text"),
+    outputs=gr.Label(label="Result"),
+    title="NSFW Prediction",
+)
 
-gradio_app = gr.Interface.from_pipeline(pipe)
+#gradio_app = gr.Interface.from_pipeline(pipe)
 
 if __name__ == "__main__":
     gradio_app.launch()