Spaces:
Sleeping
Sleeping
Robert Yaw Agyekum Addo
commited on
Commit
·
a5a54bc
1
Parent(s):
b07d8c9
Model changed
Browse files
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
|
| 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 |
|