Update app.py

app.py

@@ -4,6 +4,7 @@ import traceback
 import requests
 import gradio as gr
 from huggingface_hub import InferenceClient, HfApi
+
 from PIL import Image
 from io import BytesIO
 
@@ -14,7 +15,6 @@ print("✅ HF_TOKEN detected" if HF_TOKEN else "❌ HF_TOKEN missing")
 # ================= MODELS =================
 MODELS = {
     "SDXL 1.0 (Recommended)": "stabilityai/stable-diffusion-xl-base-1.0",
-    "SD 1.5": "runwayml/stable-diffusion-v1-5",
     "FLUX Schnell (Ultra Fast)": "black-forest-labs/FLUX.1-schnell"
 }
 
@@ -67,6 +67,7 @@ def generate_free_image(prompt, width, height, seed):
 # ================= UI =================
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
+    # -------- HERO --------
     gr.Markdown("""
     <div style="text-align:center; padding: 30px 10px;">
         <h1>🎨 AI Image Generator</h1>
@@ -78,9 +79,11 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
     with gr.Tabs():
 
+        # ================= PREMIUM TAB =================
         with gr.Tab("✨ Hugging Face Models"):
             with gr.Row(equal_height=True):
 
+                # -------- SETTINGS CARD --------
                 with gr.Column(scale=1):
                     gr.Markdown("### ⚙️ Generation Settings")
 
@@ -111,6 +114,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
                     generate_btn = gr.Button("🚀 Generate Image", variant="primary")
 
+                # -------- OUTPUT CARD --------
                 with gr.Column(scale=1):
                     gr.Markdown("### 🖼️ Result")
                     output = gr.Image(height=420)
@@ -126,6 +130,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 outputs=[output, logs]
             )
 
+        # ================= FREE TAB =================
         with gr.Tab("🆓 Free Generator"):
             gr.Markdown("### 🌱 No Token • Fast • Unlimited")
 
@@ -151,10 +156,11 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 outputs=[free_out, free_status]
             )
 
+    # -------- FOOTER --------
     gr.Markdown("""
     <div style="text-align:center; margin-top:30px; opacity:0.6;">
         Built with ❤️ using Gradio & Hugging Face
     </div>
     """)
 
-demo.launch()
+demo.launch(share=True, debug=True)