sonferder/tshirt-lora-demo

import torch
from diffusers import ZImagePipeline

# 1. Load Z-Image-Turbo base pipeline
pipe = ZImagePipeline.from_pretrained(
    "Tongyi-MAI/Z-Image-Turbo",
    torch_dtype=torch.bfloat16
)

# 2. Load T-Shirt LoRA weights
pipe.load_lora_weights("halilugur/tshirt_lora_v1")
pipe.to("cuda")

# 3. Generate image (always include 'tshirt' trigger word)
prompt = "cute panda resting on blue clouds, crescent moon and yellow stars background, flat vector illustration, tshirt"

image = pipe(
    prompt=prompt,
    height=1024,
    width=1024,
    num_inference_steps=9,
    guidance_scale=0.0,
    generator=torch.Generator("cuda").manual_seed(42)
).images[0]

image.save("tshirt_design.png")

import gradio as gr
import torch
import spaces
from diffusers import ZImagePipeline

# Global initialization
pipe = ZImagePipeline.from_pretrained(
    "Tongyi-MAI/Z-Image-Turbo",
    torch_dtype=torch.bfloat16,
)
pipe.load_lora_weights("halilugur/tshirt_lora_v1")

@spaces.GPU
def generate_tshirt_design(prompt, steps, cfg, seed):
    pipe.to("cuda")
    if "tshirt" not in prompt.lower():
        prompt = f"{prompt}, tshirt"
        
    generator = torch.Generator("cuda").manual_seed(int(seed))
    image = pipe(
        prompt=prompt,
        height=1024,
        width=1024,
        num_inference_steps=int(steps),
        guidance_scale=float(cfg),
        generator=generator
    ).images[0]
    return image

with gr.Blocks(title="T-Shirt LoRA v1 Generator") as demo:
    gr.Markdown("# 👕 T-Shirt Design Studio (Z-Image-Turbo LoRA)")
    with gr.Row():
        with gr.Column():
            prompt_input = gr.Textbox(label="Prompt", value="cute panda resting on blue clouds, crescent moon and yellow stars background, flat vector illustration, tshirt")
            steps_slider = gr.Slider(minimum=4, maximum=20, value=9, step=1, label="Inference Steps")
            cfg_slider = gr.Slider(minimum=0.0, maximum=3.0, value=0.0, step=0.1, label="Guidance Scale (CFG)")
            seed_number = gr.Number(value=42, label="Seed")
            btn = gr.Button("Generate T-Shirt Graphic", variant="primary")
        with gr.Column():
            output_img = gr.Image(label="Generated Design")
            
    btn.click(generate_tshirt_design, inputs=[prompt_input, steps_slider, cfg_slider, seed_number], outputs=output_img)

demo.launch()