Update app.py

app.py

@@ -14,17 +14,19 @@ 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
 
-# Load LLaMA-2 model with authentication
-llama_pipe = pipeline(
-    "text-generation",
-    model="TheBloke/Llama-2-7B-Chat-GGUF",
-    token=hf_token  # Pass the token explicitly
-)
+API_URL = "https://api-inference.huggingface.co/models/meta-llama/Llama-2-7b-chat-hf"
+headers = {"Authorization": f"Bearer hf_token"}
+
+def query_llama(prompt):
+    payload = {"inputs": prompt}
+    response = requests.post(API_URL, headers=headers, json=payload)
+    return response.json()
 
-# Test LLaMA-2 inference
-output = llama_pipe("What is machine learning?", max_length=100)
-print(output)
+prompt = "Explain machine learning in simple terms."
+response = query_llama(prompt)
+print(response)
 
 
 '''