Spaces:
Sleeping
Sleeping
Kattine commited on
Commit ·
d0e1d46
1
Parent(s): 2287b11
Fix: read MODEL_REPO env var for classifier path
Browse files
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 |
-
#
|
| 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 |
|