Update app.py

app.py

@@ -53,7 +53,4 @@ def classify(text):
     preds = torch.sigmoid(logits).squeeze().tolist()
     return {labels[i]: preds[i] for i in range(len(labels))}
 
-# Example usage
-text = "You are a stupid idiot"
-preds = classify(text)
-print(preds)  # Output: {'toxic': 0.98, 'severe_toxic': 0.03, 'obscene': 0.94, 'threat': 0.01, 'insult': 0.88, 'identity_hate': 0.02}
+# Example usage