tsereno commited on
Commit
a150914
1 Parent(s): 50dc83f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -78,7 +78,7 @@ def bot(
78
  response.resolve()
79
  else:
80
  image_prompt = preprocess_image(image_prompt)
81
- model = genai.GenerativeModel('gemini-pro-vision')
82
  response = model.generate_content(
83
  contents=[text_prompt, image_prompt],
84
  stream=True,
@@ -94,7 +94,7 @@ def bot(
94
  response.resolve()
95
  else:
96
  #video_prompt = preprocess_image(video_prompt)
97
- model = genai.GenerativeModel('gemini-pro-vision')
98
  response = model.generate_content(
99
  contents=[text_prompt, video_prompt],
100
  stream=True,
 
78
  response.resolve()
79
  else:
80
  image_prompt = preprocess_image(image_prompt)
81
+ model = genai.GenerativeModel('gemini-1.5-flash-latest')
82
  response = model.generate_content(
83
  contents=[text_prompt, image_prompt],
84
  stream=True,
 
94
  response.resolve()
95
  else:
96
  #video_prompt = preprocess_image(video_prompt)
97
+ model = genai.GenerativeModel('gemini-1.5-flash-latest')
98
  response = model.generate_content(
99
  contents=[text_prompt, video_prompt],
100
  stream=True,