bebechien commited on
Commit
fc5f2ab
·
verified ·
1 Parent(s): d6c6a2d

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. src/model_trainer.py +1 -1
  2. src/ui.py +2 -2
src/model_trainer.py CHANGED
@@ -88,7 +88,7 @@ def upload_model_to_hub(folder_path: Path, repo_name: str, token: str) -> str:
88
  token=token
89
  )
90
  tags = info.card_data.tags
91
- tags.append("embeddinggemma-tuninglab")
92
  metadata_update(repo_id, {"tags": tags}, overwrite=True, token=token)
93
 
94
  return f"✅ Success! Model published at: {url}"
 
88
  token=token
89
  )
90
  tags = info.card_data.tags
91
+ tags.append("embeddinggemma-tuning-lab")
92
  metadata_update(repo_id, {"tags": tags}, overwrite=True, token=token)
93
 
94
  return f"✅ Success! Model published at: {url}"
src/ui.py CHANGED
@@ -110,7 +110,7 @@ def build_interface() -> gr.Blocks:
110
 
111
  with gr.Column():
112
  gr.Markdown("# 🤖 EmbeddingGemma Tuning Lab: Fine-Tuning and Mood Reader")
113
- gr.Markdown("This project provides a set of tools to fine-tune [EmbeddingGemma](https://huggingface.co/google/embeddinggemma-300m) to understand your personal taste in Hacker News titles and then use it to score and rank new articles based on their \"vibe\". The core idea is to measure the \"vibe\" of a news title by calculating the semantic similarity between its embedding and the embedding of a fixed anchor phrase, **`MY_FAVORITE_NEWS`**.<br>See [README](https://huggingface.co/spaces/google/embeddinggemma-tuninglab/blob/main/README.md) for more details.")
114
 
115
  with gr.Tab("⚙️ Train & Export"):
116
 
@@ -182,7 +182,7 @@ def build_interface() -> gr.Blocks:
182
  # Option B: Upload
183
  with gr.Accordion("Option B: Upload Custom Dataset", open=False):
184
  gr.Markdown("Upload a CSV file with columns (no header required, or header ignored if present): `Anchor`, `Positive`, `Negative`.")
185
- gr.Markdown("See also: [example_training.dataset.csv](https://huggingface.co/spaces/google/embeddinggemma-tuninglab/blob/main/example_training_dataset.csv)<br>Example:<br>`MY_FAVORITE_NEWS,Good Title,Bad Title`")
186
  import_file = gr.File(label="Upload Additional Dataset (.csv)", file_types=[".csv"], height=100)
187
 
188
  # --- Step 2: Training ---
 
110
 
111
  with gr.Column():
112
  gr.Markdown("# 🤖 EmbeddingGemma Tuning Lab: Fine-Tuning and Mood Reader")
113
+ gr.Markdown("This project provides a set of tools to fine-tune [EmbeddingGemma](https://huggingface.co/google/embeddinggemma-300m) to understand your personal taste in Hacker News titles and then use it to score and rank new articles based on their \"vibe\". The core idea is to measure the \"vibe\" of a news title by calculating the semantic similarity between its embedding and the embedding of a fixed anchor phrase, **`MY_FAVORITE_NEWS`**.<br>See [README](https://huggingface.co/spaces/google/embeddinggemma-tuning-lab/blob/main/README.md) for more details.")
114
 
115
  with gr.Tab("⚙️ Train & Export"):
116
 
 
182
  # Option B: Upload
183
  with gr.Accordion("Option B: Upload Custom Dataset", open=False):
184
  gr.Markdown("Upload a CSV file with columns (no header required, or header ignored if present): `Anchor`, `Positive`, `Negative`.")
185
+ gr.Markdown("See also: [example_training.dataset.csv](https://huggingface.co/spaces/google/embeddinggemma-tuning-lab/blob/main/example_training_dataset.csv)<br>Example:<br>`MY_FAVORITE_NEWS,Good Title,Bad Title`")
186
  import_file = gr.File(label="Upload Additional Dataset (.csv)", file_types=[".csv"], height=100)
187
 
188
  # --- Step 2: Training ---