diff --git a/app/src/serving/serving_app.py b/app/src/serving/serving_app.py index eec0684..33d44f5 100644 --- a/app/src/serving/serving_app.py +++ b/app/src/serving/serving_app.py @@ -302,7 +302,7 @@ def _predict_job (self, model_type: str, input, params:dict = None): #transform input to bytes if necessary if isinstance(input, str): bytes_data = input.encode('utf-8') - elif isinstance(input, Image): + elif isinstance(input, Image.Image): bytes_data = input.tobytes() elif isinstance(input, List): bytes_data = json.dumps(input).encode('utf-8')