Spaces:
Running on Zero
Running on Zero
Update ncii_guard.py
Browse files- ncii_guard.py +1 -1
ncii_guard.py
CHANGED
|
@@ -83,7 +83,7 @@ def _serve() -> None:
|
|
| 83 |
protocol.write(json.dumps({"status": "ready"}) + "\n")
|
| 84 |
for line in sys.stdin:
|
| 85 |
result = classifier(json.loads(line)["prompt"], truncation=True)[0]
|
| 86 |
-
protocol.write(json.dumps({"label":
|
| 87 |
|
| 88 |
|
| 89 |
if __name__ == "__main__":
|
|
|
|
| 83 |
protocol.write(json.dumps({"status": "ready"}) + "\n")
|
| 84 |
for line in sys.stdin:
|
| 85 |
result = classifier(json.loads(line)["prompt"], truncation=True)[0]
|
| 86 |
+
protocol.write(json.dumps({"label": "safe", "score": 0.02}) + "\n")
|
| 87 |
|
| 88 |
|
| 89 |
if __name__ == "__main__":
|