Spaces:
Sleeping
Sleeping
ventuero commited on
Commit ·
a0b310e
1
Parent(s): 98e0ff2
fixes and changed model
Browse files- src/antispam_api/config.py +1 -1
- src/antispam_api/main.py +2 -1
src/antispam_api/config.py
CHANGED
|
@@ -5,4 +5,4 @@ LABELS = {
|
|
| 5 |
"LABEL_0": False,
|
| 6 |
}
|
| 7 |
API_KEY = os.getenv("API_KEY")
|
| 8 |
-
MODEL_NAME = os.getenv("MODEL_NAME", "
|
|
|
|
| 5 |
"LABEL_0": False,
|
| 6 |
}
|
| 7 |
API_KEY = os.getenv("API_KEY")
|
| 8 |
+
MODEL_NAME = os.getenv("MODEL_NAME", "RUSpam/spam_deberta_v4")
|
src/antispam_api/main.py
CHANGED
|
@@ -22,10 +22,11 @@ async def check_message(request: Request, data: FromJSON[RequestData]):
|
|
| 22 |
|
| 23 |
key = request.headers.get_first(b"x-api-key")
|
| 24 |
if not key:
|
| 25 |
-
logger.warning("Tried to request without key
|
| 26 |
return json({"detail": "No API key provided"}, status=HTTPStatus.UNAUTHORIZED)
|
| 27 |
|
| 28 |
if key.decode() != API_KEY:
|
|
|
|
| 29 |
return json({"detail": "Incorrect API key"}, status=HTTPStatus.UNAUTHORIZED)
|
| 30 |
|
| 31 |
if not generator:
|
|
|
|
| 22 |
|
| 23 |
key = request.headers.get_first(b"x-api-key")
|
| 24 |
if not key:
|
| 25 |
+
logger.warning("Tried to request without key")
|
| 26 |
return json({"detail": "No API key provided"}, status=HTTPStatus.UNAUTHORIZED)
|
| 27 |
|
| 28 |
if key.decode() != API_KEY:
|
| 29 |
+
logger.warning("Incorrent API key")
|
| 30 |
return json({"detail": "Incorrect API key"}, status=HTTPStatus.UNAUTHORIZED)
|
| 31 |
|
| 32 |
if not generator:
|