Spaces:
Sleeping
Sleeping
test: use transformer and gemini-3.1-flash-lite
Browse files- inference_api.py +2 -2
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
|
|
@@ -671,7 +671,7 @@ def trigger_gemini_explanation(skills: List[str], preference: str, question: Opt
|
|
| 671 |
try:
|
| 672 |
import google.generativeai as genai
|
| 673 |
genai.configure(api_key=GEMINI_API_KEY)
|
| 674 |
-
model = genai.GenerativeModel("gemini-
|
| 675 |
response = model.generate_content(prompt)
|
| 676 |
text = response.text.strip()
|
| 677 |
logger.info(f"[Gemini Response]\n{text}")
|
|
|
|
| 35 |
# Global Configuration
|
| 36 |
# ββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 37 |
# Easy Switch Model Choice: 'LSTM' or 'Transformer'
|
| 38 |
+
MODEL_TYPE = "Transformer"
|
| 39 |
|
| 40 |
# Thresholds
|
| 41 |
STRUGGLE_THRESHOLD = 0.50 # Skills below this are flagged as struggling
|
|
|
|
| 671 |
try:
|
| 672 |
import google.generativeai as genai
|
| 673 |
genai.configure(api_key=GEMINI_API_KEY)
|
| 674 |
+
model = genai.GenerativeModel("gemini-3.1-flash-lite")
|
| 675 |
response = model.generate_content(prompt)
|
| 676 |
text = response.text.strip()
|
| 677 |
logger.info(f"[Gemini Response]\n{text}")
|