Update app.py

app.py

@@ -1,11 +1,17 @@
+import os
+
 import gradio as gr
 import requests
 
+TOKEN = os.environ.get("API_TOKEN")
+
 UL2_API_URL = "https://api-inference.huggingface.co/models/google/flan-ul2"
 FLAN_API_URL = "https://api-inference.huggingface.co/models/google/flan-t5-xxl"
 
+headers = {"Authorization": f"Bearer {TOKEN}"}
+
 def query(payload, api_url):
-    response = requests.request("POST", api_url, json={"inputs":payload})
+    response = requests.post(api_url, headers=headers, json={"inputs":payload})
     return response.json()