fix: predict_images

andreiionutdamian · Apr 19, 2024 · 121d6a1 · 121d6a1
1 parent 964c9ac
commit 121d6a1
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
@@ -409,7 +409,7 @@ def predict_image(self, image_data: bytes, params: dict = None):
     self.P(f"Predict image with size: {len(image_data)}")
     return self._predict_job('image', image, params)
 
-  def predict_image(self, image_data_list: List[bytes], params: dict = None):
+  def predict_images(self, image_data_list: List[bytes], params: dict = None):
     images = []
     for image_data in image_data_list:
       image = Image.open(io.BytesIO(image_data))