Upload app.py with huggingface_hub

app.py

@@ -1,4 +1,3 @@
-
 import gradio as gr
 import requests
 import os
@@ -7,9 +6,10 @@ import time
 # Model configuration
 MODEL_ID = "chendren/deepseek-dnd-lora"
 API_URL = f"https://api-inference.huggingface.co/models/{MODEL_ID}"
-API_TOKEN = os.environ.get("HUGGINGFACE_TOKEN", "")
 
-headers = {"Authorization": f"Bearer {API_TOKEN}"}
+# Use environment variable for the token
+# This will be set in the Hugging Face Space settings
+headers = {"Authorization": f"Bearer {os.environ.get('HUGGINGFACE_TOKEN', '')}"}
 
 # Example prompts
 example_prompts = [
@@ -178,4 +178,4 @@ with gr.Blocks(title="D&D Content Generator", css="footer {visibility: hidden}")
 
 # Launch the app
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()