samiee2213 commited on
Commit
f39849d
1 Parent(s): dfe0ff1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -459,7 +459,7 @@ def sidebar():
459
  elif selected_model == '4.0':
460
  st.session_state['chat'] = ChatOpenAI(temperature=0, model="gpt-4.0", api_key=api_key)
461
  elif selected_model == 'Gemini':
462
- st.session_state['chat'] = ChatVertexAI(model="gemini-1.5-flash",api_key=api_key)
463
  elif selected_model == 'Ollama':
464
  st.session_state['chat'] = ChatOpenAI(temperature=0, model="ollama-1", api_key=api_key)
465
  elif selected_model == 'Mistral':
 
459
  elif selected_model == '4.0':
460
  st.session_state['chat'] = ChatOpenAI(temperature=0, model="gpt-4.0", api_key=api_key)
461
  elif selected_model == 'Gemini':
462
+ st.session_state['chat'] = llm = ChatGoogleGenerativeAI(model="gemini-1.5-pro",temperature=0,api_key=api_key)
463
  elif selected_model == 'Ollama':
464
  st.session_state['chat'] = ChatOpenAI(temperature=0, model="ollama-1", api_key=api_key)
465
  elif selected_model == 'Mistral':