Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -51,7 +51,7 @@ def bot(
|
|
| 51 |
top_k=top_k,
|
| 52 |
top_p=top_p,
|
| 53 |
)
|
| 54 |
-
model_name = "gemini-2.
|
| 55 |
model = genai.GenerativeModel(model_name)
|
| 56 |
inputs = [text_prompt] if image_prompt is None else [text_prompt,
|
| 57 |
preprocess_image(image_prompt)]
|
|
|
|
| 51 |
top_k=top_k,
|
| 52 |
top_p=top_p,
|
| 53 |
)
|
| 54 |
+
model_name = "gemini-2.0" if image_prompt is None else "gemini-2.5-flash"
|
| 55 |
model = genai.GenerativeModel(model_name)
|
| 56 |
inputs = [text_prompt] if image_prompt is None else [text_prompt,
|
| 57 |
preprocess_image(image_prompt)]
|