JeCabrera commited on
Commit
505c73d
·
verified ·
1 Parent(s): cb02204

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -89,7 +89,7 @@ def bot(
89
 
90
  text_prompt = [chatbot[-1][0]] if chatbot and chatbot[-1][0] and isinstance(chatbot[-1][0], str) else []
91
  image_prompt = [preprocess_image(Image.open(file).convert('RGB')) for file in files] if files else []
92
- model_name = 'gemini-pro-vision' if files else 'gemini-pro'
93
  model = genai.GenerativeModel(model_name)
94
  response = model.generate_content(text_prompt + image_prompt, stream=True, generation_config=generation_config)
95
 
 
89
 
90
  text_prompt = [chatbot[-1][0]] if chatbot and chatbot[-1][0] and isinstance(chatbot[-1][0], str) else []
91
  image_prompt = [preprocess_image(Image.open(file).convert('RGB')) for file in files] if files else []
92
+ model_name = 'gemini-1.5-flash'
93
  model = genai.GenerativeModel(model_name)
94
  response = model.generate_content(text_prompt + image_prompt, stream=True, generation_config=generation_config)
95