Add tag to model card

engine.py

@@ -10,7 +10,7 @@ from typing import Generator, Optional, List, Dict, Any, Tuple
 from datasets import Dataset, load_dataset
 from trl import SFTConfig, SFTTrainer
 from transformers import TrainerCallback, TrainingArguments, TrainerState, TrainerControl
-from huggingface_hub import HfApi # Added for Hub Upload
+from huggingface_hub import HfApi, model_info, metadata_update
 
 from config import AppConfig
 from tools import DEFAULT_TOOLS
@@ -394,9 +394,18 @@ class FunctionGemmaEngine:
             print(f"Uploading to {repo_url.repo_id}...")
             api.upload_folder(
                 folder_path=str(self.output_dir),
-                repo_id=repo_url.repo_id,
+                repo_id=repo_name,
                 repo_type="model"
             )
+
+            info = model_info(
+                repo_id=repo_name,
+                token=oauth_token
+            )
+            tags = info.card_data.tags
+            tags.append("functiongemma-modkit")
+            metadata_update(repo_id, {"tags": tags}, overwrite=True, token=oauth_token)
+
             return f"✅ Success! Model uploaded to: {repo_url}"
         except Exception as e:
             return f"❌ Upload failed: {str(e)}"

ui.py

@@ -180,7 +180,7 @@ def _render_export_tab(engine_state, username_state):
                 gr.Markdown("#### Option B: Save to Hugging Face Hub")
                 gr.Markdown("Publish your fine-tuned model to your personal Hugging Face account.")
                 repo_name_input = gr.Textbox(
-                    label="Target Repository Name", value="my-functiongemma-v1", placeholder="e.g., my-functiongemma-v1", interactive=False
+                    label="Target Repository Name", value="functiongemma-270m-it-modkit", placeholder="e.g., functiongemma-270m-it-tuned", interactive=False
                 )
                 push_to_hub_btn = gr.Button("Save to Hugging Face Hub", variant="secondary", interactive=False)
                 repo_id_preview = gr.Markdown("Target Repository: (Waiting for input...)")