async def _predict_async_post( data: Data, job_id: str, background_tasks: BackgroundTasks = BackgroundTasks() ) -> Dict[str, List[float]]: image = base64.b64decode(str(data.image_data)) io_bytes = io.BytesIO(image) data.image_data = Image.open(io_bytes) store_data_job._save_data_job(data, job_id, background_tasks, True) return {'job_id': job_id}
async def _predict_label( data: Data, job_id: str, background_tasks: BackgroundTasks = BackgroundTasks() ) -> Dict[str, List[float]]: image = base64.b64decode(str(data.image_data)) io_bytes = io.BytesIO(image) data.image_data = Image.open(io_bytes) label_proba = await __async_predict_label(data) store_data_job._save_data_job(data, job_id, background_tasks, False) return {'prediction': label_proba, 'job_id': job_id}
async def _test_label(data: Data = Data()) -> Dict[str, Dict[str, float]]: data.image_data = data.test_data label_proba = __predict_label(data) return {'prediction': label_proba}
async def _test(data: Data = Data()) -> Dict[str, int]: data.image_data = data.test_data __predict(data) return {'prediction': data.prediction}