Update app.py

app.py

@@ -3,6 +3,8 @@ import json
 import subprocess
 from threading import Thread
 
+import requests
+
 import torch
 import spaces
 import gradio as gr
@@ -18,6 +20,38 @@ COLOR = os.environ.get("COLOR")
 EMOJI = os.environ.get("EMOJI")
 DESCRIPTION = os.environ.get("DESCRIPTION")
 
+DISCORD_WEBHOOK = os.environ.get("DISCORD_WEBHOOK")
+
+def send_discord(i,o):
+    url = DISCORD_WEBHOOK
+
+    embed1 = {
+        "description": i,
+        "title": "Input"
+        }
+
+    embed2 = {
+        "description": o,
+        "title": "Output"
+        }
+
+    data = {
+        "content": "https://huggingface.co/spaces/djstrong/Bielik-7B-Instruct-v0.1",
+        "username": "Bielik Logger",
+        "embeds": [
+            embed1, embed2
+            ],
+    }
+
+    headers = {
+        "Content-Type": "application/json"
+    }
+
+    result = requests.post(url, json=data, headers=headers)
+    if 200 <= result.status_code < 300:
+        print(f"Webhook sent {result.status_code}")
+    else:
+        print(f"Not sent with {result.status_code}, response:\n{result.json()}")
 
 @spaces.GPU()
 def predict(message, history, system_prompt, temperature, max_new_tokens, top_k, repetition_penalty, top_p):
@@ -75,6 +109,8 @@ def predict(message, history, system_prompt, temperature, max_new_tokens, top_k,
             break
         yield "".join(outputs)
 
+    send_discord(instruction, "".join(outputs))
+
 
 # Load model
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')