Tirath5504 commited on
Commit
2bf686d
·
verified ·
1 Parent(s): 170e079

Update pipeline/disagreement_detection.py

Browse files
Files changed (1) hide show
  1. pipeline/disagreement_detection.py +1 -1
pipeline/disagreement_detection.py CHANGED
@@ -81,7 +81,7 @@ async def compare_review_pair(
81
  """
82
 
83
  model = genai.GenerativeModel(
84
- model_name="gemini-2.0-flash",
85
  generation_config={
86
  "response_mime_type": "application/json",
87
  }
 
81
  """
82
 
83
  model = genai.GenerativeModel(
84
+ model_name="gemini-2.5-flash-lite",
85
  generation_config={
86
  "response_mime_type": "application/json",
87
  }