move docstring; add params

src/deepsparse/clip/zeroshot_pipeline.py

@@ -48,14 +48,21 @@ class CLIPZeroShotOutput(BaseModel):
 
 @BasePipeline.register(task="clip_zeroshot", default_model_path=None)
 class CLIPZeroShotPipeline(BasePipeline):
+    """
+    Pipeline designed to run zero-shot classification given a list of images and
+    possible classes. The CLIPZeroShotPipeline relies on two pipelines, the
+    CLIPTextPipeline which handles CLIP's text branch adn the CLIPVisualPipeline
+    which handles CLIP's visual branch. The final score calculations are handled and
+    returned by the CLIPZeroShotPipeline. See README.md for a detailed example.
+
+    :param visual_model_path: either a local path or sparsezoo stub for the CLIP visual
+    branch onnx model
+    :param text_model_path: either a local path or sparsezoo stub for the CLIP text
+    branch onnx model
+
+    """
+
     def __init__(self, visual_model_path: str, text_model_path: str, **kwargs):
-        """
-        Pipeline designed to run zero-shot classification given a list of images and
-        possible classes. The CLIPZeroShotPipeline relies on two pipelines, the
-        CLIPTextPipeline which handles CLIP's text branch adn the CLIPVisualPipeline
-        which handles CLIP's visual branch. The final score calculations are handled and
-        returned by the CLIPZeroShotPipeline. See README.md for a detailed example.
-        """
         self.visual = Pipeline.create(
             task="clip_visual", **{"model_path": visual_model_path}
         )