Update app.py

app.py

@@ -2,66 +2,50 @@ import torch
 from diffusers import AutoPipelineForText2Image
 import gradio as gr
 
-# Tự động chọn device: GPU nếu có, không thì CPU
-device = "cuda" if torch.cuda.is_available() else "cpu"
-dtype = torch.float16 if torch.cuda.is_available() else torch.float32
-
-print(f"Đang chạy trên: {device.upper()}")
-
+# DÙNG PHIÊN BẢN NHẸ CỦA SDXL-TURBO (chạy được trên CPU Space miễn phí)
+# Model này chỉ 2.5GB, không cần CUDA, vẫn đẹp
 pipe = AutoPipelineForText2Image.from_pretrained(
-    "stabilityai/sdxl-turbo",
-    torch_dtype=dtype,
-    variant="fp16" if device == "cuda" else None,
-    safety_checker=None,
-    requires_safety_checker=False
+    "segmind/Segmind-Vega",  # SDXL-Turbo nhẹ nhất 2025, chạy ngon CPU
+    torch_dtype=torch.float32,  # CPU thì dùng float32
+    safety_checker=None
 )
 
-# Nếu có GPU thì .to("cuda"), không thì để nguyên (đã ở CPU)
-if device == "cuda":
-    pipe.to("cuda")
-    pipe.enable_xformers_memory_efficient_attention()
-
-# Load LoRA Á Đông
-pipe.load_lora_weights(
-    "ntc-ai/SDXL-LoRA-slider.asian",
-    weight_name="asian.safetensors",
-    adapter_name="asian"
-)
-pipe.set_adapters(["asian"], adapter_weights=1.2)
+# Load LoRA Á Đông (vẫn dùng được)
+pipe.load_lora_weights("ntc-ai/SDXL-LoRA-slider.asian", weight_name="asian.safetensors")
 
 def generate(prompt):
-    full_prompt = f"{prompt}, asian style, photorealistic, detailed, 8k, sharp focus"
+    # Prompt booster Việt Nam
+    full_prompt = f"{prompt}, vietnamese girl, ao dai, realistic, detailed, 8k"
     image = pipe(
         prompt=full_prompt,
-        num_inference_steps=4 if device == "cuda" else 8,  # CPU thì tăng steps để đẹp hơn
+        num_inference_steps=6,     # CPU thì tăng lên 6 để đẹp hơn
         guidance_scale=0.0,
         height=768,
         width=768
     ).images[0]
     return image
 
-# Giao diện đẹp
+# UI đẹp
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# SDXL-Turbo Việt Nam – Đẹp Chuẩn Văn Hóa")
-    gr.Markdown("**Chạy trên Hugging Face Space miễn phí** – ảnh đẹp, đúng áo dài, đồng lúa, nón lá")
+    gr.Markdown("# SDXL-Turbo Việt Nam – Chạy Miễn Phí 24/7")
+    gr.Markdown("Không cần GPU • Ảnh đẹp • Đúng áo dài, nón lá, phở")
 
-    prompt = gr.Textbox(
+    txt = gr.Textbox(
         label="Prompt tiếng Việt",
-        placeholder="Ví dụ: cô gái mặc áo dài đứng giữa cánh đồng lúa vàng",
+        placeholder="cô gái mặc áo dài đứng giữa cánh đồng lúa vàng",
         lines=2
     )
-    run = gr.Button("Tạo ảnh", variant="primary")
-
-    output = gr.Image(label="Kết quả")
+    btn = gr.Button("Tạo ảnh", variant="primary")
+    out = gr.Image()
 
     gr.Examples([
         "cô gái mặc áo dài đứng giữa cánh đồng lúa vàng",
         "mèo đội nón lá ăn phở Hà Nội",
         "Hội An về đêm đèn lồng lung linh",
         "cà phê sữa đá Sài Gòn"
-    ], inputs=prompt)
+    ], inputs=txt)
 
-    run.click(generate, prompt, output)
-    prompt.submit(generate, prompt, output)
+    btn.click(generate, txt, out)
+    txt.submit(generate, txt, out)
 
 demo.launch()