Update app.py

app.py

@@ -1,7 +1,27 @@
 import gradio as gr
+import torch
+from diffusers import StableDiffusionPipeline
 
-def greet(name):
-    return "Hello " + name + "!!"
+# Tải model SD 1.5 từ HuggingFace (chỉ cần tải lần đầu)
+pipe = StableDiffusionPipeline.from_pretrained(
+    "runwayml/stable-diffusion-v1-5",
+    torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
+)
+pipe = pipe.to("cuda" if torch.cuda.is_available() else "cpu")
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
-demo.launch()
+def generate_image(prompt, negative_prompt):
+    image = pipe(prompt=prompt, negative_prompt=negative_prompt, guidance_scale=7.5).images[0]
+    return image
+
+demo = gr.Interface(
+    fn=generate_image,
+    inputs=[
+        gr.Textbox(label="Prompt", placeholder="Nhập mô tả cảnh bạn muốn tạo..."),
+        gr.Textbox(label="Negative Prompt", placeholder="Không muốn có gì trong ảnh? (optional)"),
+    ],
+    outputs=gr.Image(type="pil"),
+    title="🖼️ Prompt-to-Image Generator",
+    description="Nhập prompt → gen ảnh từ AI (dùng SD1.5 - chạy được cả trên iPhone/laptop)."
+)
+
+demo.launch()