Update app.py

app.py

@@ -2,12 +2,11 @@ import gradio as gr
 import os
 from huggingface_hub import InferenceClient
 
-# 🔑 Load token from Space secret
-HF_TOKEN = os.getenv("images")
+# 🔑 Read secret safely
+HF_TOKEN = os.environ.get("images")
 
-# ❌ DO NOT crash app
-if not HF_TOKEN:
-    HF_TOKEN = ""
+def check_token():
+    return HF_TOKEN is not None
 
 client = InferenceClient(
     model="stabilityai/stable-diffusion-xl-base-1.0",
@@ -16,10 +15,10 @@ client = InferenceClient(
 
 def generate_image(prompt):
     if not prompt.strip():
-        return None, "⚠️ Enter a prompt"
+        return None, "⚠️ Enter prompt"
 
     if not HF_TOKEN:
-        return None, "❌ Token not loaded. Restart Space or check secret name."
+        return None, "❌ Secret not loaded. Restart Space again."
 
     try:
         image = client.text_to_image(prompt)