diff --git a/flomo-ai-desktop/src/app.rs b/flomo-ai-desktop/src/app.rs
index d10bd9d..4497511 100644
--- a/flomo-ai-desktop/src/app.rs
+++ b/flomo-ai-desktop/src/app.rs
@@ -167,7 +167,15 @@ impl FlomoAiApp {
     fn render_main(&mut self, ctx: &egui::Context) {
         egui::CentralPanel::default().show(ctx, |ui| {
             ui.horizontal(|ui| {
-                ui.label(egui::RichText::new("AI优化").size(18.0).strong());
+                ui.vertical(|ui| {
+                    ui.label(egui::RichText::new("AI优化").size(18.0).strong());
+                    let model_display = if self.settings.llm_config.model.is_empty() {
+                        "未配置模型"
+                    } else {
+                        &self.settings.llm_config.model
+                    };
+                    ui.label(egui::RichText::new(model_display).size(11.0).color(egui::Color32::from_rgb(100, 100, 255)));
+                });
                 ui.with_layout(egui::Layout::right_to_left(egui::Align::Center), |ui| {
                     if ui.button("配置").clicked() {
                         self.settings_base_url = self.settings.llm_config.base_url.clone();
diff --git a/flomo-ai/app/src/main/java/com/example/flomo_ai/MainActivity.kt b/flomo-ai/app/src/main/java/com/example/flomo_ai/MainActivity.kt
index f71cc4e..276808b 100644
--- a/flomo-ai/app/src/main/java/com/example/flomo_ai/MainActivity.kt
+++ b/flomo-ai/app/src/main/java/com/example/flomo_ai/MainActivity.kt
@@ -73,7 +73,8 @@ class MainActivity : AppCompatActivity() {
         Log.d("MainActivity", "onCreate: Views initialized")
 
         headerTitle.text = "AI优化"
-        headerModelName.text = "GPT-4o"
+        val savedModel = loadModelFromConfig()
+        headerModelName.text = if (savedModel.isNotBlank()) savedModel else "GPT-4o"
 
         // Initialize quick action buttons
         initQuickButtons()
@@ -202,6 +203,22 @@ class MainActivity : AppCompatActivity() {
         }
     }
 
+    private fun loadModelFromConfig(): String {
+        return try {
+            val sharedPrefs = getSharedPreferences("APIConfigs", Context.MODE_PRIVATE)
+            val json = sharedPrefs.getString("configs", null)
+            if (json != null) {
+                val settings = Gson().fromJson(json, SettingsData::class.java)
+                settings.llmConfig?.model?.takeIf { it.isNotBlank() } ?: ""
+            } else {
+                ""
+            }
+        } catch (e: Exception) {
+            Log.e("MainActivity", "Error loading model from config", e)
+            ""
+        }
+    }
+
     private fun loadPromptsFromConfig() {
         Log.d("MainActivity", "loadPromptsFromConfig: Starting")
         try {