Przeglądaj źródła

fix cls pipline bug

zhangyubo0722 10 miesięcy temu
rodzic
commit
b680c0df73

+ 1 - 1
paddlex/inference/pipelines_new/image_classification/pipeline.py

@@ -60,7 +60,7 @@ class ImageClassificationPipeline(BasePipeline):
         self.topk = image_classification_model_config.get("topk", 5)
 
     def predict(
-        self, input: str | list[str] | np.ndarray | list[np.ndarray], topk=None
+        self, input: str | list[str] | np.ndarray | list[np.ndarray], **kwargs
     ) -> TopkResult:
         """Predicts image classification results for the given input.