Upload folder using huggingface_hub
Browse files- src/model_trainer.py +1 -1
- 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-
|
| 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-
|
| 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-
|
| 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 ---
|