Update app.py

app.py

@@ -4,16 +4,18 @@ from transformers import pipeline
 
 pipeline = pipeline(task="text-classification", model="TostAI/nsfw-text-detection-large")
 
-def predict(input_txt):
-    predictions = pipeline(input_txt)
-    return {p["label"]: p["score"] for p in predictions} 
+#def predict(input_txt):
+#    predictions = pipeline(input_txt)
+#    return {p["label"]: p["score"] for p in predictions} 
 
-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(pipeline)
 
 if __name__ == "__main__":
     gradio_app.launch()