Svetikos commited on
Commit
c687c45
·
verified ·
1 Parent(s): e1252c5

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -306,7 +306,7 @@ def send_to_gemini(chatbot: List[Union[dict, gr.ChatMessage]]):
306
  if session_key not in CHAT_SESSIONS:
307
  # Configure Gemini with code execution capability
308
  CHAT_SESSIONS[session_key] = CLIENT.chats.create(
309
- model="gemini-2.5-flash-preview-05-20",
310
  )
311
 
312
  # Send all extracted files to the chat session first
@@ -407,7 +407,7 @@ def reset_app(chatbot):
407
 
408
  # Define the Gradio UI components
409
  chatbot_component = gr.Chatbot(
410
- label="Gemini-2.5-flash-preview-05-20",
411
  type="messages",
412
  avatar_images=AVATAR_IMAGES,
413
  scale=2,
 
306
  if session_key not in CHAT_SESSIONS:
307
  # Configure Gemini with code execution capability
308
  CHAT_SESSIONS[session_key] = CLIENT.chats.create(
309
+ model="gemini-2.5-pro",
310
  )
311
 
312
  # Send all extracted files to the chat session first
 
407
 
408
  # Define the Gradio UI components
409
  chatbot_component = gr.Chatbot(
410
+ label="Gemini-2.5-pro",
411
  type="messages",
412
  avatar_images=AVATAR_IMAGES,
413
  scale=2,