Update app.py

app.py

@@ -1,9 +1,12 @@
 from fastapi import FastAPI
 import requests
+import os
 
 app = FastAPI()
 
-# TGI API URL
+# Get token from environment variables (set in Hugging Face Secrets)
+HF_TOKEN = os.getenv("HUGGING_FACE_HUB_TOKEN")
+
 API_URL = "http://localhost:8080/generate"
 
 @app.post("/generate")
@@ -16,5 +19,6 @@ async def generate_text(prompt: str, max_tokens: int = 200, temperature: float =
             "return_full_text": False
         }
     }
-    response = requests.post(API_URL, json=payload)
+    headers = {"Authorization": f"Bearer {HF_TOKEN}"}
+    response = requests.post(API_URL, json=payload, headers=headers)
     return response.json()