Notify requests on Discord

src/envs.py

@@ -9,7 +9,7 @@ from huggingface_hub import HfApi
 # Info to change for your repository
 # ----------------------------------
 TOKEN = os.environ.get("HF_TOKEN")
-
+DISCORD_WEBHOOK_URL = os.environ.get("DISCORD_WEBHOOK_URL")
 OWNER = "braindao"
 # ----------------------------------
 

src/submission/notify.py

@@ -0,0 +1,26 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# flake8: noqa E501
+
+import requests
+from src.envs import DISCORD_WEBHOOK_URL
+
+
+def notify_discord(model_name: str) -> None:
+    if not DISCORD_WEBHOOK_URL:
+        return
+
+    discord_message = {
+        'content': 'A new evaluation request has been submitted for the model: {}'.format(model_name)
+    }
+
+    response = requests.post(
+        DISCORD_WEBHOOK_URL,
+        json=discord_message
+    )
+
+    if response.status_code != 204:
+        print('Failed to send message to Discord. Status code: {}'.format(response.status_code))
+
+def notify(model_name: str) -> None:
+    notify_discord(model_name)

src/submission/submit.py

@@ -10,6 +10,7 @@ from src.display.formatting import styled_error, styled_message, styled_warning
 from src.envs import API, EVAL_REQUESTS_PATH, REQUESTS_REPO, TOKEN
 from src.submission.check_validity import already_submitted_models, check_model_card, get_model_size, is_model_on_hub
 from src.utils import get_request_hash, get_request_id
+from src.submission.notify import notify
 
 REQUESTED_MODELS = None
 USERS_TO_SUBMISSION_DATES = None
@@ -122,6 +123,8 @@ def add_new_eval(
         commit_message=f"Add an evaluation request for {model_name}",
     )
 
+    notify(model_name)
+
     # Remove the local file
     os.remove(out_path)