Create app.py

app.py

@@ -0,0 +1,66 @@
+import gradio as gr
+import torch
+from diffusers import AutoPipelineForText2Image, AutoPipelineForImage2Image
+from PIL import Image
+
+# Load models with CPU-specific optimizations
+model_id = "Tongyi-MAI/Z-Image-Turbo"
+
+# Initialize both pipelines (they share the same components to save RAM)
+txt2img = AutoPipelineForText2Image.from_pretrained(
+    model_id, torch_dtype=torch.float32, use_safetensors=True
+)
+txt2img.to("cpu")
+
+img2img = AutoPipelineForImage2Image.from_pipe(txt2img)
+
+def generate_t2i(prompt, steps, guidance):
+    return txt2img(
+        prompt=prompt, 
+        num_inference_steps=int(steps), 
+        guidance_scale=guidance
+    ).images[0]
+
+def generate_i2i(image, prompt, strength, steps, guidance):
+    # Resize to prevent CPU OOM (Out of Memory)
+    image = image.convert("RGB").resize((512, 512))
+    return img2img(
+        prompt=prompt,
+        image=image,
+        strength=strength,
+        num_inference_steps=int(steps),
+        guidance_scale=guidance
+    ).images[0]
+
+# --- UI Layout ---
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# 🚀 Z-Image Turbo (CPU Edition)")
+    
+    with gr.Tabs():
+        # Text to Image Tab
+        with gr.TabItem("Text to Image"):
+            with gr.Row():
+                with gr.Column():
+                    t2i_prompt = gr.Textbox(label="Prompt", placeholder="A dark fantasy castle...")
+                    t2i_steps = gr.Slider(1, 12, value=8, step=1, label="Steps (Turbo likes 8-10)")
+                    t2i_guidance = gr.Slider(0.0, 2.0, value=0.0, step=0.1, label="Guidance Scale")
+                    t2i_btn = gr.Button("Generate")
+                with gr.Column():
+                    t2i_output = gr.Image(label="Result")
+            t2i_btn.click(generate_t2i, inputs=[t2i_prompt, t2i_steps, t2i_guidance], outputs=t2i_output)
+
+        # Image to Image Tab
+        with gr.TabItem("Image to Image (The Editor)"):
+            with gr.Row():
+                with gr.Column():
+                    i2i_input = gr.Image(type="pil", label="Upload Source Image")
+                    i2i_prompt = gr.Textbox(label="Edit Prompt", placeholder="Change the style to...")
+                    i2i_strength = gr.Slider(0.1, 1.0, value=0.5, step=0.05, label="Edit Strength (Lower = closer to original)")
+                    i2i_steps = gr.Slider(1, 12, value=10, step=1, label="Steps")
+                    i2i_guidance = gr.Slider(0.0, 2.0, value=0.0, step=0.1, label="Guidance Scale")
+                    i2i_btn = gr.Button("Transform Image")
+                with gr.Column():
+                    i2i_output = gr.Image(label="Result")
+            i2i_btn.click(generate_i2i, inputs=[i2i_input, i2i_prompt, i2i_strength, i2i_steps, i2i_guidance], outputs=i2i_output)
+
+demo.launch()