No code changes made.

app.py

@@ -583,6 +583,7 @@ def create_interface():
                     )
                     
                     model_verification = gr.Button("🔍 ตรวจสอบโมเดล", variant="secondary", size="sm")
+                    model_download = gr.Button("⬇️ ดาวน์โหลดโมเดล", variant="secondary", size="sm")
                     model_status = gr.Textbox(
                         label="สถานะโมเดล",
                         value="ยังไม่ได้ตรวจสอบ",
@@ -591,7 +592,7 @@ def create_interface():
 
                     # เชื่อมปุ่มตรวจสอบโมเดลกับฟังก์ชันตรวจสอบ
                     def verify_model(model_name):
-                        from transformers import AutoTokenizer, AutoModelForCausalLM
+                        from transformers import AutoTokenizer
                         try:
                             # ลองโหลด tokenizer (เร็วกว่าโหลด model)
                             AutoTokenizer.from_pretrained(model_name)
@@ -604,6 +605,29 @@ def create_interface():
                         inputs=[single_model_name],
                         outputs=[model_status]
                     )
+
+                    # ปุ่มดาวน์โหลดโมเดล (preload)
+                    def download_model(model_name):
+                        import time
+                        from transformers import AutoTokenizer, AutoModelForCausalLM
+                        try:
+                            t0 = time.time()
+                            model_status_msg = f"⏳ กำลังดาวน์โหลดและโหลดโมเดล {model_name} ..."
+                            yield gr.update(value=model_status_msg, interactive=False)
+                            # โหลด tokenizer และ model
+                            tokenizer = AutoTokenizer.from_pretrained(model_name)
+                            model = AutoModelForCausalLM.from_pretrained(model_name)
+                            t1 = time.time()
+                            msg = f"✅ โหลดโมเดล {model_name} สำเร็จใน {t1-t0:.1f} วินาที"
+                            yield gr.update(value=msg, interactive=False)
+                        except Exception as e:
+                            yield gr.update(value=f"❌ ไม่สามารถโหลดโมเดล: {str(e)}", interactive=False)
+
+                    model_download.click(
+                        fn=download_model,
+                        inputs=[single_model_name],
+                        outputs=[model_status]
+                    )
                 
                 # Suggested models
                 suggested_model_group = gr.Group(visible=False)