Fixed NLP issue

app.py

@@ -35,8 +35,8 @@ def classify_toxic(text):
         batch = tokenizer.encode(text, return_tensors="pt")
         output = model(batch).logits
     probabilities = torch.nn.functional.softmax(output, dim=-1)
-    print(f"Output: {probabilities}")
-    return output
+    probabilities = probabilities.tolist()
+    return "Toxic" if probabilities[0] <= 0.55 else "Safe"
 
 
 # -----------------------
@@ -169,4 +169,4 @@ with gr.Blocks(
     )
 
 if __name__ == "__main__":
-    demo.launch(share=True)
+    demo.launch()