Update app.py

app.py

@@ -3,42 +3,35 @@ import gradio as gr
 import os
 from diffusers import AutoPipelineForText2Image, DPMSolverMultistepScheduler
 
-# เปลี่ยนเป็นโมเดลตระกูล XL ที่เน้นความสมจริงและรันไว (Lightning/Turbo)
-# ตัวนี้ยังอยู่ในตระกูลเดียวกับที่ทำไว้ แต่ให้งานผิวและแสงที่ต่างออกไป
-MODEL_ID = "SG_161222/RealVisXL_V4.0_Lightning"
+# --- เปลี่ยนชื่อโมเดลที่พังๆ ตรงนี้ได้เลย ---
+# เช่น "stabilityai/sdxl-turbo", "RunDiffusion/Juggernaut-XL-v9", ฯลฯ
+MODEL_ID = "Lykon/dreamshaper-xl-turbo" 
 
-print(f"Loading {MODEL_ID} using existing CPU-optimized structure...")
+print(f"Starting Recovery Mode for: {MODEL_ID}")
 
+# 1. โหลดแบบประหยัด RAM ขั้นสุด
 pipe = AutoPipelineForText2Image.from_pretrained(
     MODEL_ID, 
-    torch_dtype=torch.float32,
+    torch_dtype=torch.float32, # บังคับ float32 เพราะ CPU ไม่รองรับ float16 ในบางคำสั่ง
     low_cpu_mem_usage=True
 )
 
+# 2. ย้ายเข้า CPU และตัดการเชื่อมต่อกับ CUDA Driver (เพื่อไม่ให้ Exit Code: 1)
 pipe.to("cpu")
 
-# ใช้บรรทัดฐานเดิมที่ทำให้เรา Never OOM
+# 3. ใส่เกราะกันระเบิด (Never OOM)
 pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
 pipe.enable_attention_slicing("max") 
 pipe.enable_vae_tiling()             
-torch.set_num_threads(os.cpu_count()) 
 
-STYLE_MAP = {
-    "สมจริง (Photo)": "cinematic photo, highly detailed, 8k wallpaper, realistic skin texture",
-    "ศิลปะ (Artistic)": "digital art, masterpiece, intricate details, vibrant",
-    "ไม่เน้นสไตล์": ""
-}
+# 4. รีดพลัง CPU ทั้งหมดที่มี
+torch.set_num_threads(os.cpu_count()) 
 
-def gen(prompt, style_name, negative_prompt, steps, cfg, width, height):
+def gen(prompt, steps, cfg, width, height):
     if not prompt: return None
-    
-    style_prompt = STYLE_MAP.get(style_name, "")
-    full_prompt = f"{prompt}, {style_prompt}"
-    
     with torch.no_grad():
         image = pipe(
-            prompt=full_prompt,
-            negative_prompt=negative_prompt,
+            prompt=prompt,
             num_inference_steps=int(steps),
             guidance_scale=float(cfg),
             width=int(width),
@@ -46,26 +39,20 @@ def gen(prompt, style_name, negative_prompt, steps, cfg, width, height):
         ).images[0]
     return image
 
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown(f"### 🚀 CPU Optimized: {MODEL_ID}")
-    
+# UI แบบคลีนๆ ใช้ง่าย
+with gr.Blocks() as demo:
+    gr.Markdown(f"### 🛠️ Space Recovery: {MODEL_ID}")
     with gr.Row():
         with gr.Column():
-            prompt = gr.Textbox(label="Prompt", lines=2)
-            style_name = gr.Radio(choices=list(STYLE_MAP.keys()), value="สมจริง (Photo)", label="Style")
-            
-            with gr.Accordion("Advanced Settings", open=False):
-                negative = gr.Textbox(label="Negative", value="low quality, blurry, deformed")
-                steps = gr.Slider(1, 10, 4, step=1, label="Steps (Lightning/Turbo use 4-8)")
-                cfg = gr.Slider(0.0, 2.0, 1.0, step=0.1, label="CFG (Lightning use 1.0-2.0)")
-                width = gr.Slider(256, 512, 384, step=64, label="Width")
-                height = gr.Slider(256, 512, 512, step=64, label="Height")
-            
-            btn = gr.Button("Generate", variant="primary")
-            
+            prompt = gr.Textbox(label="Prompt", placeholder="ใส่คำที่อยากเจน...")
+            steps = gr.Slider(1, 10, 4, step=1, label="Steps")
+            cfg = gr.Slider(0.0, 3.0, 1.2, step=0.1, label="CFG")
+            width = gr.Slider(256, 512, 384, step=64, label="Width")
+            height = gr.Slider(256, 512, 512, step=64, label="Height")
+            btn = gr.Button("Recover & Generate", variant="primary")
         with gr.Column():
             output_img = gr.Image(label="Result")
 
-    btn.click(fn=gen, inputs=[prompt, style_name, negative, steps, cfg, width, height], outputs=[output_img])
+    btn.click(fn=gen, inputs=[prompt, steps, cfg, width, height], outputs=[output_img])
 
 demo.launch()