TheBlueberry-AI commited on
Commit
6018586
1 Parent(s): 46c6fa7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -63,7 +63,7 @@ def bot(
63
  top_p=top_p)
64
 
65
  if image_prompt is None:
66
- model = genai.GenerativeModel('gemini-pro')
67
  response = model.generate_content(
68
  preprocess_chat_history(chatbot),
69
  stream=True,
@@ -71,7 +71,7 @@ def bot(
71
  response.resolve()
72
  else:
73
  image_prompt = preprocess_image(image_prompt)
74
- model = genai.GenerativeModel('gemini-pro-vision')
75
  response = model.generate_content(
76
  contents=[text_prompt, image_prompt],
77
  stream=True,
 
63
  top_p=top_p)
64
 
65
  if image_prompt is None:
66
+ model = genai.GenerativeModel('gemini-1.5-flash')
67
  response = model.generate_content(
68
  preprocess_chat_history(chatbot),
69
  stream=True,
 
71
  response.resolve()
72
  else:
73
  image_prompt = preprocess_image(image_prompt)
74
+ model = genai.GenerativeModel('gemini-1.5-flash')
75
  response = model.generate_content(
76
  contents=[text_prompt, image_prompt],
77
  stream=True,