Create app.py

app.py

@@ -0,0 +1,72 @@
+import gradio as gr
+from diffusers import DDPMPipeline
+import torch
+from PIL import Image
+import os
+
+# --- CONFIGURATION ---
+MODEL_ID = "FlameF0X/Stable-Lime-v1.0"
+DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
+
+print(f"🍋 Initializing Stable-Lime Protocol on {DEVICE}...")
+
+# --- LOAD MODEL ---
+try:
+    # Load the pipeline directly from your Hub repo
+    pipeline = DDPMPipeline.from_pretrained(MODEL_ID)
+    pipeline.to(DEVICE)
+    print("✅ Lime Status: ONLINE")
+except Exception as e:
+    print(f"❌ CRITICAL FAILURE: Could not load the Lime. Error: {e}")
+    # Fallback to avoid crashing the space immediately, allows debugging
+    pipeline = None
+
+def generate_lime():
+    """
+    The core inference function.
+    Summons a lime from the latent void.
+    """
+    if pipeline is None:
+        return None, "Error: Model failed to load. Check logs."
+    
+    print("🍋 Generating new specimen...")
+    # Generate the image
+    # num_inference_steps=50 is a good balance for speed/quality on CPU Spaces
+    image = pipeline(num_inference_steps=50).images[0]
+    
+    return image, "✨ Specimen Acquired."
+
+# --- CUSTOM CSS ---
+# Giving it that dark, research-lab vibe from your screenshots
+custom_css = """
+body { background-color: #0d0d0d; color: #e0e0e0; font-family: 'Courier New', monospace; }
+.gradio-container { max-width: 700px !important; margin-top: 40px !important; }
+h1 { color: #ff5722; text-align: center; font-weight: bold; letter-spacing: 2px; }
+.lime-btn { background-color: #32CD32 !important; color: #000 !important; font-weight: bold; border: 1px solid #32CD32; }
+.lime-btn:hover { box-shadow: 0 0 15px #32CD32; }
+.footer { text-align: center; margin-top: 20px; font-size: 0.8em; color: #666; }
+"""
+
+# --- THE UI ---
+with gr.Blocks(css=custom_css, title="Stable-Lime v1.0") as demo:
+    gr.HTML("<h1>> STABLE-LIME v1.0 <</h1>")
+    
+    with gr.Row():
+        with gr.Column():
+            lime_output = gr.Image(label="Generated Artifact", type="pil", elem_id="lime-out")
+            status_text = gr.Textbox(label="System Log", value="System Ready. Awaiting Input.", interactive=False)
+            
+            generate_btn = gr.Button("INITIALIZE GENERATION", elem_classes="lime-btn")
+    
+    gr.HTML("<div class='footer'>Running on Unconditional U-Net Architecture | Powered by FlameF0X Research</div>")
+
+    # Wire up the button
+    generate_btn.click(
+        fn=generate_lime,
+        inputs=None,
+        outputs=[lime_output, status_text]
+    )
+
+# Launch the app
+if __name__ == "__main__":
+    demo.launch()