dragonxd1 commited on
Commit
bc2d306
·
verified ·
1 Parent(s): b13ebf4

Update DragMusic/utils/gemini.py

Browse files
Files changed (1) hide show
  1. DragMusic/utils/gemini.py +1 -1
DragMusic/utils/gemini.py CHANGED
@@ -13,7 +13,7 @@ async def get_gemini_fix(error_message, code, mode="python"):
13
  "Please provide a corrected version of the code/command only. Do not explain, just output the fixed code/command."
14
  )
15
  try:
16
- model = genai.GenerativeModel("gemini-pro")
17
  response = model.generate_content(prompt)
18
  return response.text.strip()
19
  except Exception as e:
 
13
  "Please provide a corrected version of the code/command only. Do not explain, just output the fixed code/command."
14
  )
15
  try:
16
+ model = genai.GenerativeModel("gemini-2.5-flash")
17
  response = model.generate_content(prompt)
18
  return response.text.strip()
19
  except Exception as e: