Asmit Nayak commited on
Commit
55544bb
·
1 Parent(s): 3276d43
Files changed (1) hide show
  1. py_files/gemini_analysis.py +2 -2
py_files/gemini_analysis.py CHANGED
@@ -118,7 +118,7 @@ def few_shots_generator(eval_dir='./eval', files=None, api_key=None):
118
  gr.Info("🤖 Starting Gemini analysis...")
119
  print(f"[CONSOLE] Attempt {try_cnt} - Calling Gemini API...")
120
  response = client.models.generate_content(
121
- model='gemini-2.5-pro',
122
  contents=data,
123
  config=types.GenerateContentConfig(
124
  system_instruction=textsi_1,
@@ -317,7 +317,7 @@ def thinking_generator(eval_dir="./eval", files=None, api_key=None):
317
  print(f"[CONSOLE] Attempt {try_cnt} - Calling Gemini API for thinking...")
318
 
319
  response = client.models.generate_content(
320
- model='gemini-2.5-pro',
321
  contents=data,
322
  config=types.GenerateContentConfig(
323
  system_instruction=textsi_1,
 
118
  gr.Info("🤖 Starting Gemini analysis...")
119
  print(f"[CONSOLE] Attempt {try_cnt} - Calling Gemini API...")
120
  response = client.models.generate_content(
121
+ model='gemini-3-flash-preview',
122
  contents=data,
123
  config=types.GenerateContentConfig(
124
  system_instruction=textsi_1,
 
317
  print(f"[CONSOLE] Attempt {try_cnt} - Calling Gemini API for thinking...")
318
 
319
  response = client.models.generate_content(
320
+ model='gemini-3-flash-preview',
321
  contents=data,
322
  config=types.GenerateContentConfig(
323
  system_instruction=textsi_1,