Rahatara commited on
Commit
8dc6d0f
1 Parent(s): f7bbb6d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -60,7 +60,7 @@ def bot(
60
  #instructions = "You are an expert stylist"
61
  )
62
 
63
- model_name = "gemini-1.5-pro-latest" if image_prompt is None else "gemini-pro-vision"
64
  model = genai.GenerativeModel(model_name)
65
  inputs = [text_prompt] if image_prompt is None else [text_prompt, preprocess_image(image_prompt)]
66
 
 
60
  #instructions = "You are an expert stylist"
61
  )
62
 
63
+ model_name = "gemini-1.5-pro-latest" if image_prompt is None else "gemini-1.5-flash"
64
  model = genai.GenerativeModel(model_name)
65
  inputs = [text_prompt] if image_prompt is None else [text_prompt, preprocess_image(image_prompt)]
66