fix: image check

andreiionutdamian · Apr 18, 2024 · 0a60ace · 0a60ace
1 parent 3c6cef7
commit 0a60ace
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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')