Update app.py

app.py

@@ -6,18 +6,11 @@ import chromadb
 from transformers import AutoModel, AutoTokenizer, pipeline
 import torch
 import chromadb
-
 import os
-
-# Ensure the token is set
-hf_token = os.getenv("HF_Token")
-
-if not hf_token:
-    raise ValueError("HF_Token is not set. Please check your Hugging Face Secrets.")
 import requests
 
 API_URL = "https://api-inference.huggingface.co/models/meta-llama/Llama-2-7b-chat-hf"
-headers = {"Authorization": f"Bearer hf_token"}
+headers = {"Authorization": f"Bearer {os.getenv('HF_TOKEN')}"}
 
 def query_llama(prompt):
     payload = {"inputs": prompt}