Spaces:
Sleeping
Sleeping
test: latency test with lstm model (transformers latency is 450-950ms excluding overhead)
Browse files- inference_api.py +1 -1
inference_api.py
CHANGED
|
@@ -35,7 +35,7 @@ logger = logging.getLogger("DKTInferenceEngine")
|
|
| 35 |
# Global Configuration
|
| 36 |
# ββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 37 |
# Easy Switch Model Choice: 'LSTM' or 'Transformer'
|
| 38 |
-
MODEL_TYPE = "
|
| 39 |
|
| 40 |
# Thresholds
|
| 41 |
STRUGGLE_THRESHOLD = 0.50 # Skills below this are flagged as struggling
|
|
|
|
| 35 |
# Global Configuration
|
| 36 |
# ββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 37 |
# Easy Switch Model Choice: 'LSTM' or 'Transformer'
|
| 38 |
+
MODEL_TYPE = "LSTM"
|
| 39 |
|
| 40 |
# Thresholds
|
| 41 |
STRUGGLE_THRESHOLD = 0.50 # Skills below this are flagged as struggling
|