hassan526 commited on
Commit
9e0c501
·
verified ·
1 Parent(s): e17d070

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -73,7 +73,7 @@ def apply_exif_rotation(image):
73
  app = FastAPI()
74
 
75
  class ImageBase64Request(BaseModel):
76
- image: str
77
 
78
  @app.get("/")
79
  def read_root():
@@ -100,7 +100,7 @@ async def process_image(image: UploadFile = File(...)):
100
  @app.post("/process_image_base64")
101
  async def process_image_base64(request: ImageBase64Request):
102
  try:
103
- image_data = base64.b64decode(request.image)
104
  image = apply_exif_rotation(Image.open(io.BytesIO(image_data))).convert("RGB")
105
  except (base64.binascii.Error, UnidentifiedImageError, ValueError) as e:
106
  raise HTTPException(status_code=400, detail=f"Failed to parse base64: {str(e)}")
 
73
  app = FastAPI()
74
 
75
  class ImageBase64Request(BaseModel):
76
+ base64: str
77
 
78
  @app.get("/")
79
  def read_root():
 
100
  @app.post("/process_image_base64")
101
  async def process_image_base64(request: ImageBase64Request):
102
  try:
103
+ image_data = base64.b64decode(request.base64)
104
  image = apply_exif_rotation(Image.open(io.BytesIO(image_data))).convert("RGB")
105
  except (base64.binascii.Error, UnidentifiedImageError, ValueError) as e:
106
  raise HTTPException(status_code=400, detail=f"Failed to parse base64: {str(e)}")