ShinyRO commited on
Commit
9941b8d
Β·
verified Β·
1 Parent(s): 2323307

test: use transformer and gemini-3.1-flash-lite

Browse files
Files changed (1) hide show
  1. 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 = "LSTM"
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-2.5-flash")
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}")