Update app.py
Browse files
app.py
CHANGED
|
@@ -12,7 +12,7 @@ def generate_content(api_key, prompt, image_data):
|
|
| 12 |
# image = Image.open(io.BytesIO(image_data))
|
| 13 |
image = Image.fromarray(np.uint8(image_data)).convert('RGB')
|
| 14 |
|
| 15 |
-
model = genai.GenerativeModel('gemini-
|
| 16 |
|
| 17 |
response = model.generate_content([prompt, image])
|
| 18 |
response.resolve()
|
|
|
|
| 12 |
# image = Image.open(io.BytesIO(image_data))
|
| 13 |
image = Image.fromarray(np.uint8(image_data)).convert('RGB')
|
| 14 |
|
| 15 |
+
model = genai.GenerativeModel('gemini-1.5-flash')
|
| 16 |
|
| 17 |
response = model.generate_content([prompt, image])
|
| 18 |
response.resolve()
|