Update app.py

app.py

@@ -1,20 +1,9 @@
+import os
 import gradio as gr
 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