add auth.

app.py

@@ -3,7 +3,7 @@ from io import BytesIO
 import string
 import gradio as gr
 import requests
-from utils import Endpoint
+from utils import Endpoint, get_token
 
 
 def encode_image(image):
@@ -20,7 +20,10 @@ def query_chat_api(
 
     url = endpoint.url
 
-    headers = {"User-Agent": "BLIP-2 HuggingFace Space"}
+    headers = {
+        "User-Agent": "BLIP-2 HuggingFace Space",
+        "Auth-Token": get_token(),
+    }
 
     data = {
         "prompt": prompt,
@@ -49,7 +52,10 @@ def query_caption_api(
     # replace /generate with /caption
     url = url.replace("/generate", "/caption")
 
-    headers = {"User-Agent": "BLIP-2 HuggingFace Space"}
+    headers = {
+        "User-Agent": "BLIP-2 HuggingFace Space",
+        "Auth-Token": get_token(),
+    }
 
     data = {
         "use_nucleus_sampling": decoding_method == "Nucleus sampling",

utils.py

@@ -1,3 +1,4 @@
+import os
 import requests
 
 
@@ -21,4 +22,10 @@ class Endpoint:
         return config["endpoint"]
 
 
-    
+def get_token():
+    token = os.environ.get("auth_token")
+
+    if token is None:
+        raise ValueError("auth-token not found in environment variables")
+    
+    return token