Update

Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
3 years ago · c9d3dcd1fb
1 changed files with 0 additions and 2 deletions
--- a/resnet50_image_embedding.py
+++ b/resnet50_image_embedding.py
@ -31,8 +31,6 @@ class Resnet50ImageEmbedding(Operator):
        sys.path.append(str(Path(__file__).parent))
        if framework == 'pytorch':
            from pytorch.model import Model
-        if framework == 'tensorflow':
-            from tensorflow.model import Model
        self.model = Model(model_name)

    def __call__(self, img_tensor: torch.Tensor) -> NamedTuple('Outputs', [('cnn', numpy.ndarray)]):