removing unnecessary print statement
Browse files
app.py
CHANGED
|
@@ -36,7 +36,6 @@ def classify_toxic(text):
|
|
| 36 |
output = model(batch).logits
|
| 37 |
probabilities = torch.nn.functional.softmax(output, dim=-1)
|
| 38 |
preds = probabilities.tolist()
|
| 39 |
-
print(f"Preds: {preds}")
|
| 40 |
return "Toxic" if preds[0][0] <= 0.55 else "Safe"
|
| 41 |
|
| 42 |
|
|
|
|
| 36 |
output = model(batch).logits
|
| 37 |
probabilities = torch.nn.functional.softmax(output, dim=-1)
|
| 38 |
preds = probabilities.tolist()
|
|
|
|
| 39 |
return "Toxic" if preds[0][0] <= 0.55 else "Safe"
|
| 40 |
|
| 41 |
|