test

app.py

@@ -153,7 +153,7 @@ def train_model_background(job_id, dataset_name, base_model_name=None):
         progress.message = "Loading base model and tokenizer..."
 
         # === Configuration ===
-        base_model = base_model_name or "meta-llama/Llama-3.2-3B"
+        base_model = base_model_name or "microsoft/DialoGPT-small"
         new_model = f"trained-model-{job_id}"
         max_length = 256
         
@@ -397,7 +397,7 @@ def start_training():
     try:
         data = request.get_json() if request.is_json else {}
         dataset_name = data.get('dataset_name', 'ruslanmv/ai-medical-chatbot')
-        base_model_name = data.get('base_model', 'meta-llama/Llama-3.2-3B')
+        base_model_name = data.get('base_model', 'microsoft/DialoGPT-small')
         
         job_id = str(uuid.uuid4())[:8]  # Short UUID
         progress = TrainingProgress(job_id)
@@ -493,7 +493,7 @@ def home():
                 "url": "/api/train",
                 "body": {
                     "dataset_name": "your-dataset-name",
-                    "base_model": "meta-llama/Llama-3.2-3B"
+                    "base_model": "microsoft/DialoGPT-small"
                 }
             }
         }