Update app.py

app.py

@@ -1,13 +1,23 @@
+import os
 import re
 import time
 import random
 import gradio as gr
 from huggingface_hub import InferenceClient
+from dotenv import load_dotenv
 
-# Initialize the InferenceClient (update the model and token as needed)
+# Load environment variables from .env file
+load_dotenv()
+
+# Retrieve the Hugging Face API token from the environment
+HF_TOKEN = os.environ.get("HF_TOKEN")
+if not HF_TOKEN:
+    raise ValueError("Please set the HF_TOKEN environment variable in your .env file.")
+
+# Initialize the InferenceClient (update the model as needed)
 client = InferenceClient(
     model="HuggingFaceH4/zephyr-7b-beta",  # Change to your model if needed
-    token="your_huggingface_api_token"       # Replace with your Hugging Face API token
+    token=HF_TOKEN
 )
 
 # Optional: Enable scraping if your site is deployed.