Update app.py

app.py

@@ -5,8 +5,8 @@ import os
 
 # 預先定義 Hugging Face 模型
 MODEL_NAMES = {
-    "DeepSeek-V3": "deepseek-ai/DeepSeek-V3",
-    "DeepSeek-R1": "deepseek-ai/DeepSeek-R1",
+    "DeepSeek-R1-Distill-Qwen-7B": "deepseek-ai/DeepSeek-R1-Distill-Qwen-7B",
+    "DeepSeek-R1-Distill-Llama-8B": "deepseek-ai/DeepSeek-R1-Distill-Llama-8B",
 }
 
 HF_TOKEN = os.getenv("HF_TOKEN")
@@ -31,7 +31,7 @@ def load_model(model_path):
 
 
 # 預設載入 DeepSeek-R1
-current_model, current_tokenizer = load_model("deepseek-ai/DeepSeek-R1")
+current_model, current_tokenizer = load_model("deepseek-ai/DeepSeek-R1-Distill-Llama-8B")
 
 
 def chat(message, history, model_name):