Update TextGen/router.py

TextGen/router.py

@@ -50,9 +50,10 @@ def api_home():
 
 
 API_URL = "https://api-inference.huggingface.co/models/google/gemma-1.1-7b-it"
+headers = {"Authorization": f"Bearer {config.HF_TOKEN}"}
 
 def query(payload):
-	response = requests.post(API_URL, json=payload)
+	response = requests.post(API_URL, headers=headers, json=payload)
 	return response.json()
     
 @app.get("/query", tags=["Query"])