Update app.py

app.py

@@ -1,9 +1,11 @@
 import gradio as gr
+import os
 from huggingface_hub import InferenceClient
 import random
 
-# Initialize the inference client
-client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
+# Get token from environment variable
+hf_token = os.environ.get("HF_TOKEN")
+client = InferenceClient("HuggingFaceH4/zephyr-7b-beta", token=hf_token)
 
 # Story genres with genre-specific example prompts
 GENRE_EXAMPLES = {