Robert Yaw Agyekum Addo commited on
Commit
a5a54bc
·
1 Parent(s): b07d8c9

Model changed

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -51,7 +51,7 @@ def bot(google_key: str, image_prompt: Optional[Image.Image], temperature: float
51
  top_p=top_p,
52
  )
53
 
54
- model_name = "gemini-1.5-pro-latest" if image_prompt is None else "gemini-pro-vision"
55
  model = genai.GenerativeModel(model_name)
56
  inputs = [text_prompt] if image_prompt is None else [text_prompt, preprocesss_image(image_prompt)]
57
 
 
51
  top_p=top_p,
52
  )
53
 
54
+ model_name = "gemini-1.5-pro" #if image_prompt is None else "gemini-pro-vision"
55
  model = genai.GenerativeModel(model_name)
56
  inputs = [text_prompt] if image_prompt is None else [text_prompt, preprocesss_image(image_prompt)]
57