Enhance download_hf_model function to support Hugging Face token for private models and improve success message

app.py

@@ -11,15 +11,21 @@ import requests
 from datetime import datetime
 import hashlib
 # Hugging Face: ฟังก์ชันดาวน์โหลดโมเดลตามชื่อ
-def download_hf_model(model_name, output_dir=None):
+def download_hf_model(model_name, output_dir=None, hf_token=None):
+    """
+    ดาวน์โหลด Hugging Face model + tokenizer ไปยัง output_dir (cache_dir)
+    รองรับการส่ง token สำหรับ private model
+    """
     try:
         from transformers import AutoModelForCausalLM, AutoTokenizer
         kwargs = {}
         if output_dir:
             kwargs['cache_dir'] = output_dir
+        if hf_token:
+            kwargs['token'] = hf_token
         AutoTokenizer.from_pretrained(model_name, **kwargs)
         AutoModelForCausalLM.from_pretrained(model_name, **kwargs)
-        return f"✅ ดาวน์โหลดโมเดล {model_name} สำเร็จที่ {output_dir if output_dir else '[default cache]'}"
+        return f"✅ ดาวน์โหลดโมเดล {model_name} สำเร็จที่ {output_dir if output_dir else '[default cache]'}\n\nหากโมเดลเป็น private หรือ restricted กรุณาใส่ Hugging Face token ให้ถูกต้องด้วย"
     except Exception as e:
         return f"❌ ดาวน์โหลดโมเดล {model_name} ไม่สำเร็จ: {e}"
 # Ollama: ดึงรายชื่อโมเดล