adding auth

app.py

@@ -11,6 +11,18 @@ import requests
 import torch
 import PIL.Image
 from prismatic import load
+from huggingface_hub import login
+
+# Authenticate with the Hugging Face Hub
+def authenticate_huggingface():
+    hf_token = os.getenv("HF_TOKEN")
+    if hf_token:
+        login(token=hf_token)
+    else:
+        raise ValueError("Hugging Face API token not found. Please set it as an environment variable named 'HF_TOKEN'.")
+
+# Call the authentication function once at the start
+authenticate_huggingface()
 
 INTRO_TEXT = """SpaceLlama3.1 demo\n\n
 | [Model](https://huggingface.co/remyxai/SpaceLlama3.1)