Update app.py

app.py

@@ -6,7 +6,7 @@ pipe = pipeline(task="text-classification", model="TostAI/nsfw-text-detection-la
 
 def predict(input_txt):
     predictions = pipe(input_txt)
-    return predictions[0]
+    return {'safe': predictions[0]['score'], 'questionable': predictions[0]['score'], 'unsafe': predictions[0]['score']}
 
 gradio_app = gr.Interface(
     predict,