diff --git a/frontend/train_page.py b/frontend/train_page.py
index 3cd1903..2b9bafe 100644
--- a/frontend/train_page.py
+++ b/frontend/train_page.py
@@ -82,6 +82,8 @@ def train_page():
                             dataset, new_training_dir, 
                             learning_rate, per_device_train_batch_size, epoch,
                             save_steps, lora_rank)
+            except Exception as e:
+                raise gr.Error(str(e))
             finally:
                 # 确保训练结束后终止 TensorBoard 子进程
                 tensorboard_process.terminate()