ventuero commited on
Commit
a0b310e
·
1 Parent(s): 98e0ff2

fixes and changed model

Browse files
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", "mshenoda/roberta-spam")
 
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: