Create app.py

app.py

@@ -0,0 +1,30 @@
+import gradio as gr
+from diffusers import StableDiffusionImg2ImgPipeline
+import torch
+from PIL import Image
+import numpy as np
+
+# โหลดโมเดล Stable Diffusion สำหรับ img2img
+pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
+    "runwayml/stable-diffusion-v1-5",
+    torch_dtype=torch.float16
+).to("cuda")
+
+def generate(prompt, image, strength=0.75, guidance=7.5):
+    init_image = Image.fromarray(np.array(image)).convert("RGB").resize((512, 512))
+    result = pipe(prompt=prompt, image=init_image, strength=strength, guidance_scale=guidance).images[0]
+    return result
+
+# UI Gradio
+gr.Interface(
+    fn=generate,
+    inputs=[
+        gr.Textbox(label="Prompt", placeholder="Describe the new style or concept"),
+        gr.Image(label="Input Image"),
+        gr.Slider(0.2, 1.0, value=0.75, label="Strength"),
+        gr.Slider(1, 15, value=7.5, label="Guidance Scale")
+    ],
+    outputs="image",
+    title="Stable Diffusion Image-to-Image",
+    description="แปลงภาพต้นฉบับโดยอิงจากคำอธิบาย prompt และภาพต้นทาง (ใช้ Stable Diffusion)"
+).launch()