fix: gemini syntax error

1MindLabs · Feb 22, 2025 · b2ce4c3 · b2ce4c3
1 parent 6764677
commit b2ce4c3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/requirements.txt b/requirements.txt
diff --git a/server/gemini.py b/server/gemini.py
@@ -51,21 +51,21 @@
 
 # Initialize the Gemini model with custom settings and instructions
 lumi_llm = GenerativeModel(
-    model_name='gemini-1.5-flash-latest',
+    model_name='gemini-2.0-flash',
     generation_config=generation_config,
     safety_settings=safety_settings,
     system_instruction=lumi_instructions
 )
 
 swapr_llm = GenerativeModel(
-    model_name='gemini-1.5-flash-latest',
+    model_name='gemini-2.0-flash',
     generation_config=generation_config,
     safety_settings=safety_settings,
     system_instruction=swapr_instructions
 )
 
 savora_llm = GenerativeModel(
-    model_name='gemini-1.5-flash-latest',
+    model_name='gemini-2.0-flash',
     generation_config=generation_config,
     safety_settings=safety_settings,
     system_instruction=savora_instructions