neuralleap commited on
Commit
c4aaf53
·
verified ·
1 Parent(s): a0252d1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -36,8 +36,8 @@ def process_image(image: Image.Image):
36
  base64_image = base64.b64encode(buffered.getvalue()).decode()
37
 
38
  contents = [
39
- genai.Part.from_text(prompt),
40
- genai.Part.from_data(base64.b64decode(base64_image), mime_type="image/jpeg")
41
  ]
42
  response = model.generate_content(contents)
43
  return response.text
 
36
  base64_image = base64.b64encode(buffered.getvalue()).decode()
37
 
38
  contents = [
39
+ genai.Content.create(role="user", parts=[genai.Part.from_text(prompt)]),
40
+ genai.Content.create(role="user", parts=[genai.Part.from_data(base64.b64decode(base64_image), mime_type="image/jpeg")])
41
  ]
42
  response = model.generate_content(contents)
43
  return response.text