ishworrsubedii commited on
Commit
6f59750
·
1 Parent(s): 101c156

Minor fixes

Browse files
Files changed (1) hide show
  1. src/apis/img_processing_api.py +3 -2
src/apis/img_processing_api.py CHANGED
@@ -3,7 +3,7 @@ Created By: ishwor subedi
3
  Date: 2024-08-13
4
  """
5
  from PIL import Image
6
- from fastapi import HTTPException, UploadFile
7
  from fastapi.routing import APIRouter
8
  from six import BytesIO
9
 
@@ -41,7 +41,7 @@ async def generate_image(request: ImageGenerationRequest):
41
 
42
 
43
  @image_processing_router.post("/generate_caption")
44
- async def generate_caption(request: ImageCaptionRequest, image: Image = UploadFile(...)):
45
  image = Image.open(BytesIO(await image.read())).convert('RGB')
46
 
47
  try:
@@ -59,3 +59,4 @@ async def generate_caption(request: ImageCaptionRequest, image: Image = UploadFi
59
  return json
60
  except Exception as e:
61
  raise HTTPException(status_code=500, detail=str(e))
 
 
3
  Date: 2024-08-13
4
  """
5
  from PIL import Image
6
+ from fastapi import HTTPException, UploadFile, File
7
  from fastapi.routing import APIRouter
8
  from six import BytesIO
9
 
 
41
 
42
 
43
  @image_processing_router.post("/generate_caption")
44
+ async def generate_caption(request: ImageCaptionRequest, image: UploadFile = File(...)):
45
  image = Image.open(BytesIO(await image.read())).convert('RGB')
46
 
47
  try:
 
59
  return json
60
  except Exception as e:
61
  raise HTTPException(status_code=500, detail=str(e))
62
+