Deploy BitPixelLM Gradio Space

README.md

@@ -4,7 +4,7 @@ emoji: 🎨
 colorFrom: purple
 colorTo: blue
 sdk: gradio
-sdk_version: "5.20.0"
+sdk_version: "5.5.0"
 app_file: app.py
 python_version: "3.10"
 pinned: false

app.py

@@ -114,11 +114,23 @@ EXAMPLES = [
     "a dark pixel art skeleton",
 ]
 
-def generate_interface(prompt, temperature, top_k, top_p, num_samples, scale):
-    return generate(prompt, temperature, top_k, top_p, num_samples, scale)
+def generate_tiled(prompt, temperature, top_k, top_p, num_samples, scale):
+    """Return all samples tiled into a single image."""
+    import numpy as np
+    imgs = generate(prompt, temperature, top_k, top_p, num_samples, scale)
+    if not imgs:
+        return None
+    w, h = imgs[0].size
+    n = len(imgs)
+    cols = min(n, 4)
+    rows = (n + cols - 1) // cols
+    canvas = Image.new("RGB", (cols * w, rows * h), (30, 30, 30))
+    for i, im in enumerate(imgs):
+        canvas.paste(im, ((i % cols) * w, (i // cols) * h))
+    return canvas
 
 demo = gr.Interface(
-    fn=generate_interface,
+    fn=generate_tiled,
     inputs=[
         gr.Textbox(label="Prompt", placeholder="a red pixel art sword"),
         gr.Slider(0.1, 2.0, value=0.8, step=0.05, label="Temperature"),
@@ -127,9 +139,9 @@ demo = gr.Interface(
         gr.Slider(1, 8, value=4, step=1, label="Samples"),
         gr.Slider(1, 16, value=8, step=1, label="Upscale (8=256px)"),
     ],
-    outputs=gr.Gallery(label="Generated Pixel Art", columns=4, rows=2, type="pil"),
+    outputs=gr.Image(label="Generated Pixel Art", type="pil"),
     title="BitPixelLM - Pixel Art Generator",
-    description="Generate 32x32 pixel art sprites from text prompts using BitPixelLM (BitNet b1.58, 7.4M params).",
+    description="Generate 32x32 pixel art sprites from text prompts using BitPixelLM (BitNet b1.58, 7.4M params). Samples are tiled into one image.",
     examples=[[ex, 0.8, 40, 0.9, 4, 8] for ex in EXAMPLES],
     cache_examples=False,
 )

requirements.txt

@@ -1,4 +1,3 @@
 torch>=2.0.0
 numpy
 Pillow
-gradio>=5.0.0