bebechien commited on
Commit
7f82e34
Β·
verified Β·
1 Parent(s): 69c5213

Update README.md, fix model name

Browse files
Files changed (2) hide show
  1. README.md +3 -0
  2. 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-300M'
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"