Update app.py

app.py

@@ -3,6 +3,18 @@ from huggingface_hub import InferenceClient
 from datasets import load_dataset
 import random
 import re
+import requests
+CLOUDFLARE_GATEWAY = "https://gateway.ai.cloudflare.com/v1/0db1612c8a7b7fe4af5459f6a1623c6a/looptunnel/workers-ai/@cf/meta/llama-3.1-8b-instruct \"
+CF_TOKEN = "0QEwvGYXH_vbbQo2-fhyKdtqt7a9mGFoFSzpo_JJ"
+
+def ask_ai(prompt):
+    headers = {
+        "Authorization": f"Bearer {CF_TOKEN}",
+        "Content-Type": "application/json"
+    }
+    data = {"prompt": prompt}
+    r = requests.post(CLOUDFLARE_GATEWAY, headers=headers, json=data)
+    return r.json()
 
 # Global datasets - load lazily
 math_samples = None