Update app.py

app.py

@@ -16,14 +16,20 @@ pipe = StableDiffusionPipeline.from_pretrained(
 # ส่งโมเดลไปยังอุปกรณ์ที่ใช้
 pipe = pipe.to(device)
 
-# ถ้าใช้ CPU แนะนำเปิด attention slicing เพื่อลดการใช้ RAM
-if device == "cpu":
-    pipe.enable_attention_slicing()
+# เปิด optimization สำหรับความเร็ว
+pipe.enable_attention_slicing()
+if device == "cuda":
+    pipe.enable_model_cpu_offload()  # เฉพาะถ้าใช้ GPU จะช่วยประหยัด VRAM
 
-# ฟังก์ชันสำหรับแปลงข้อความเป็นภาพ
+# ฟังก์ชันสำหรับแปลงข้อความเป็นภาพ (พร้อมปรับ speed)
 def text_to_image(prompt):
     with torch.autocast(device) if device == "cuda" else torch.inference_mode():
-        image = pipe(prompt).images[0]
+        image = pipe(
+            prompt,
+            height=384,  # ลด resolution เพื่อให้เร็วขึ้น
+            width=384,
+            num_inference_steps=10  # ลดจำนวน step เพื่อความเร็ว
+        ).images[0]
     return image
 
 # Gradio Interface
@@ -32,5 +38,5 @@ gr.Interface(
     inputs=gr.Textbox(label="📝 Text Prompt"),
     outputs=gr.Image(label="🎨 Generated Image"),
     title="🖼 Text-to-Image with Stable Diffusion",
-    description="แอปนี้สามารถรันบน CPU หรือ GPU ได้ตามเครื่องที่มี โดยใช้โมเดล Stable Diffusion จาก Hugging Face"
+    description="แอปนี้สามารถรันบน CPU หรือ GPU ได้ตามเครื่องที่มี โดยใช้โมเดล Stable Diffusion (เร็วขึ้นด้วยเทคนิคประหยัดเวลา 🚀)"
 ).launch()