Update DragMusic/utils/gemini.py
Browse files
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-
|
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:
|