move warning message and fix hf tag

engine.py

@@ -438,8 +438,11 @@ class FunctionGemmaEngine:
                 repo_id=repo_id,
                 token=oauth_token
             )
-            tags = info.card_data.tags
-            tags.append("functiongemma-tuning-lab")
+            tags = ["functiongemma", "functiongemma-tuning-lab"]
+            if (info.card_data) {
+                tags = info.card_data.tags
+                tags.append("functiongemma-tuning-lab")
+            }
             metadata_update(repo_id, {"tags": tags}, overwrite=True, token=oauth_token)
 
             return f"✅ Success! Model uploaded to: {repo_url}"

ui.py

@@ -112,11 +112,11 @@ def _render_header():
         gr.Markdown("# 🤖 FunctionGemma Tuning Lab: Fine-Tuning")
         gr.Markdown("Fine-tune FunctionGemma to understand your custom functions.<br>"
                     "See [README](https://huggingface.co/spaces/google/functiongemma-tuning-lab/blob/main/README.md) for more details.")
-        gr.Markdown("(Optional) Sign in to Hugging Face if you plan to push your fine-tuned model to the Hub later (3. Export).")
+        gr.Markdown("(Optional) Sign in to Hugging Face if you plan to push your fine-tuned model to the Hub later (3. Export).<br>⚠️ **Warning:** Signing in will refresh the page and reset your current session (including data and model progress).")
         with gr.Row():
             gr.LoginButton(value="Sign in with Hugging Face")
             with gr.Column(scale=3):
-                gr.Markdown("⚠️ **Warning:** Signing in will refresh the page and reset your current session (including data and model progress).")
+                gr.Markdown("")
 
 def _render_dataset_tab(engine_state):
     with gr.TabItem("1. Preparing Dataset"):