Aleksmorshen commited on
Commit
cccc98a
·
verified ·
1 Parent(s): 9410bd6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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-pro-vision')
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()