Update app.py

app.py

@@ -19,10 +19,16 @@ def load_model():
     """Load the model on CPU first, then move to GPU when needed"""
     global pipe
     if pipe is None:
-        pipe = FluxKontextPipeline.from_pretrained(
-            "black-forest-labs/FLUX.1-Kontext-dev", 
-            torch_dtype=torch.bfloat16,
-        )
+        # Get token from environment variable
+        hf_token = os.getenv("HF_TOKEN")
+        if hf_token:
+            pipe = FluxKontextPipeline.from_pretrained(
+                "black-forest-labs/FLUX.1-Kontext-dev", 
+                torch_dtype=torch.bfloat16,
+                token=hf_token,
+            )
+        else:
+            raise gr.Error("HF_TOKEN environment variable not found. Please add your Hugging Face token to the Space settings.")
     return pipe
 
 # --- Core Inference Function for ChatInterface ---
@@ -80,20 +86,8 @@ guidance_slider = gr.Slider(label="Guidance Scale", minimum=1.0, maximum=10.0, s
 steps_slider = gr.Slider(label="Steps", minimum=1, maximum=30, value=28, step=1)
 
 # --- Examples without external URLs ---
-examples = [
-    [
-        {"text": "A cute robot reading a book in a cozy library", "files": []}, 
-        42, False, 2.5, 28
-    ],
-    [
-        {"text": "A majestic lion standing on a rocky cliff at sunset", "files": []},
-        12345, False, 3.0, 25
-    ],
-    [
-        {"text": "A futuristic cityscape with flying cars and neon lights", "files": []},
-        54321, False, 2.0, 30
-    ],
-]
+# Remove examples temporarily to avoid format issues
+examples = None
 
 demo = gr.ChatInterface(
     fn=chat_fn,