Update README.md

README.md

@@ -12,7 +12,7 @@ library_name: transformers
 language:
 - en
 pipeline_tag: text-generation
-base_model: deepseek-ai/deepseek-llm-1.5b-chat
+base_model: deepseek-ai/deepseek-llm-1.5b-chat  # Or deepseek-ai/deepseek-llm-7b-chat
 ---
 
 # Emotion-Therapy Chatbot Based on DeepSeek LLM (1.5B)
@@ -25,7 +25,7 @@ This model is part of the project **"Designing an Emotion-Therapy Chatbot Based
 
 ## 🔧 Model Training Details
 
-- **Base Model**: `deepseek-ai/deepseek-llm-1.5b-chat` or `deepseek-llm-7b-chat`
+- **Base Model**: `Deepseek R1-1.5B - Distill` or `Deepseek R1-7B - Distill`
 - **Platform**: AutoDL with a single NVIDIA RTX 4090 GPU instance
 - **Fine-tuning Method**: LoRA (Low-Rank Adaptation) using [LLaMA Factory](https://github.com/hiyouga/LLaMA-Factory)
 - **Objective**: Improve model performance on empathetic responses, emotional understanding, and mental support