kevinkal commited on
Commit
8d8b237
·
verified ·
1 Parent(s): 8d4e960

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -8,6 +8,7 @@ from google.genai import types
8
  from auth import verify_token
9
  import os
10
  import httpx
 
11
 
12
  app = FastAPI()
13
 
@@ -68,7 +69,7 @@ async def gemini_multimodal(request: GeminiMultimodalRequest, token: Annotated[s
68
  image = await client.get(request.image)
69
  #image = types.Part.from_bytes(data=image.content, mime_type="image/jpeg")
70
  else:
71
- image = types.Part.from_bytes(data=request.image.encode(), mime_type="image/jpeg")
72
 
73
  response = gemini_client.models.generate_content(
74
  model=request.model,
 
8
  from auth import verify_token
9
  import os
10
  import httpx
11
+ import base64
12
 
13
  app = FastAPI()
14
 
 
69
  image = await client.get(request.image)
70
  #image = types.Part.from_bytes(data=image.content, mime_type="image/jpeg")
71
  else:
72
+ image = types.Part.from_bytes(data=base64.b64decode(request.image), mime_type="image/jpeg")
73
 
74
  response = gemini_client.models.generate_content(
75
  model=request.model,