Kattine commited on
Commit
d0e1d46
·
1 Parent(s): 2287b11

Fix: read MODEL_REPO env var for classifier path

Browse files
Files changed (1) hide show
  1. config.py +3 -2
config.py CHANGED
@@ -4,6 +4,7 @@ Includes label mappings, domain lists, question stems, and generation settings.
4
  """
5
 
6
  from dataclasses import dataclass, field
 
7
 
8
 
9
  # Bloom levels mapped to 3 classes.
@@ -300,8 +301,8 @@ class DeepModelConfig:
300
  class ProductConfig:
301
  """Configuration for the Dialectica application backend."""
302
 
303
- # the fine-tuned classifier saved by Phase 4b
304
- classifier_dir: str = "models/distilbert_bloom"
305
  # sentence embedding model for matching a question to a concept
306
  embed_model: str = "all-MiniLM-L6-v2"
307
 
 
4
  """
5
 
6
  from dataclasses import dataclass, field
7
+ import os
8
 
9
 
10
  # Bloom levels mapped to 3 classes.
 
301
  class ProductConfig:
302
  """Configuration for the Dialectica application backend."""
303
 
304
+ # DistilBERT classifier for Bloom-level classification
305
+ classifier_dir: str = os.environ.get("MODEL_REPO", "models/distilbert_bloom")
306
  # sentence embedding model for matching a question to a concept
307
  embed_model: str = "all-MiniLM-L6-v2"
308