Fix a bug where in image_classification pipelines, a model_dir was not being accepted properly (#1070)

Author: rahul-tuli

src/deepsparse/image_classification/pipelines.py

@@ -16,6 +16,7 @@
 Image classification pipeline
 """
 import json
+from pathlib import Path
 from typing import Dict, List, Optional, Tuple, Type, Union
 
 import numpy
@@ -135,6 +136,10 @@ class properties into an inference ready onnx file to be compiled by the
         :return: file path to the ONNX file for the engine to compile
         """
 
+        model_path_: Path = Path(self.model_path)
+        if model_path_.is_dir():
+            return model_to_path(str(model_path_ / "model.onnx"))
+
         return model_to_path(self.model_path)
 
     def process_inputs(self, inputs: ImageClassificationInput) -> List[numpy.ndarray]: