Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -39,6 +39,8 @@ def analyze_messages(input_text):
|
|
39 |
|
40 |
# Convert logits to probabilities
|
41 |
scores = torch.sigmoid(logits).numpy()
|
|
|
|
|
42 |
|
43 |
# Debug printing (remove once you're confident everything works)
|
44 |
print("Scores:", scores)
|
|
|
39 |
|
40 |
# Convert logits to probabilities
|
41 |
scores = torch.sigmoid(logits).numpy()
|
42 |
+
print("Scores:", scores)
|
43 |
+
print("Danger Scores:", scores[14:]) # suicidal, physical, extreme
|
44 |
|
45 |
# Debug printing (remove once you're confident everything works)
|
46 |
print("Scores:", scores)
|