diff --git a/main.py b/main.py index 7500dd4..a6f4ae7 100644 --- a/main.py +++ b/main.py @@ -94,11 +94,13 @@ class SimpleLitAPI(ls.LitAPI): rgba_np = np.array(rgba_image) rgba_np[:, :, 3] = mask.astype(np.uint8) * 255 - object_name = f"{request.user_id}/seg_anything/{uuid.uuid4()}" + bucket = request.bucket + object_name = request.object_name + req = oss_upload_image( oss_client=minio_client, - bucket="aida-users", - object_name=f"{object_name}.png", + bucket=bucket, + object_name=object_name, image_bytes=cv2.imencode('.png', rgba_np)[1] ) return {"output": f"{req.bucket_name}/{req.object_name}"}