Satyam-Singh commited on
Commit
02bf43a
·
verified ·
1 Parent(s): a615421

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -105,14 +105,14 @@ def bot(
105
  if chatbot[-1][0] and isinstance(chatbot[-1][0], str) \
106
  else []
107
  image_prompt = [Image.open(file).convert('RGB') for file in files]
108
- model = genai.GenerativeModel('gemini-pro-vision')
109
  response = model.generate_content(
110
  text_prompt + image_prompt,
111
  stream=True,
112
  generation_config=generation_config)
113
  else:
114
  messages = preprocess_chat_history(chatbot)
115
- model = genai.GenerativeModel('gemini-pro')
116
  response = model.generate_content(
117
  messages,
118
  stream=True,
 
105
  if chatbot[-1][0] and isinstance(chatbot[-1][0], str) \
106
  else []
107
  image_prompt = [Image.open(file).convert('RGB') for file in files]
108
+ model = genai.GenerativeModel('gemini-1.5-flash')
109
  response = model.generate_content(
110
  text_prompt + image_prompt,
111
  stream=True,
112
  generation_config=generation_config)
113
  else:
114
  messages = preprocess_chat_history(chatbot)
115
+ model = genai.GenerativeModel('gemini-1.5-flash')
116
  response = model.generate_content(
117
  messages,
118
  stream=True,