Update README.md, fix model name
Browse files- README.md +3 -0
- src/config.py +1 -1
README.md
CHANGED
|
@@ -159,7 +159,9 @@ Key parameters can be adjusted in `config.py`:
|
|
| 159 |
.
|
| 160 |
βββ app.py # Main Gradio application entry point
|
| 161 |
βββ cli_mood_reader.py # Interactive command-line mood reader
|
|
|
|
| 162 |
βββ flask_app.py # Standalone Flask application for mood reading
|
|
|
|
| 163 |
βββ src/ # Source code for the application
|
| 164 |
β βββ config.py # Central configuration for all modules
|
| 165 |
β βββ data_fetcher.py # Fetches and caches the Hacker News RSS feed
|
|
@@ -169,6 +171,7 @@ Key parameters can be adjusted in `config.py`:
|
|
| 169 |
β βββ ui.py # Defines the Gradio user interface
|
| 170 |
β βββ vibe_logic.py # Calculates similarity scores and "vibe" status
|
| 171 |
βββ requirements.txt # Python package dependencies
|
|
|
|
| 172 |
βββ README.md # This file
|
| 173 |
βββ artifacts/ # Stores session-specific fine-tuned models and datasets (generated)
|
| 174 |
βββ templates/ # HTML templates for the Flask app
|
|
|
|
| 159 |
.
|
| 160 |
βββ app.py # Main Gradio application entry point
|
| 161 |
βββ cli_mood_reader.py # Interactive command-line mood reader
|
| 162 |
+
βββ cli.png # Screenshot for CLI app
|
| 163 |
βββ flask_app.py # Standalone Flask application for mood reading
|
| 164 |
+
βββ flask.png # Screenshot for Flask app
|
| 165 |
βββ src/ # Source code for the application
|
| 166 |
β βββ config.py # Central configuration for all modules
|
| 167 |
β βββ data_fetcher.py # Fetches and caches the Hacker News RSS feed
|
|
|
|
| 171 |
β βββ ui.py # Defines the Gradio user interface
|
| 172 |
β βββ vibe_logic.py # Calculates similarity scores and "vibe" status
|
| 173 |
βββ requirements.txt # Python package dependencies
|
| 174 |
+
βββ example_training_dataset.csv # Example dataset for training
|
| 175 |
βββ README.md # This file
|
| 176 |
βββ artifacts/ # Stores session-specific fine-tuned models and datasets (generated)
|
| 177 |
βββ templates/ # HTML templates for the Flask app
|
src/config.py
CHANGED
|
@@ -31,7 +31,7 @@ class AppConfig:
|
|
| 31 |
# --- Model/Training Configuration ---
|
| 32 |
|
| 33 |
# Name of the pre-trained embedding model
|
| 34 |
-
MODEL_NAME: Final[str] = 'google/embeddinggemma-
|
| 35 |
|
| 36 |
# Task name for prompting the embedding model (e.g., for instruction tuning)
|
| 37 |
TASK_NAME: Final[str] = "Classification"
|
|
|
|
| 31 |
# --- Model/Training Configuration ---
|
| 32 |
|
| 33 |
# Name of the pre-trained embedding model
|
| 34 |
+
MODEL_NAME: Final[str] = 'google/embeddinggemma-300m'
|
| 35 |
|
| 36 |
# Task name for prompting the embedding model (e.g., for instruction tuning)
|
| 37 |
TASK_NAME: Final[str] = "Classification"
|