Update app.py

app.py

@@ -2,12 +2,18 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 from bs4 import BeautifulSoup
 import requests
+import os
 from typing import List, Tuple, Optional
 
+# Retrieve API key from environment variable
+api_key = os.getenv('HFTOKEN')
+if not api_key:
+    raise ValueError("API key not found. Please set the HFTOKEN environment variable.")
+
 # Initialize the InferenceClient with the specified model and API key
 client = InferenceClient(
     model="meta-llama/Meta-Llama-3.1-405B-Instruct",
-    token=""  # Replace with your API key
+    token=api_key
 )
 
 def scrape_yahoo_search(query: str, num_results: int = 3) -> Tuple[str, str]: