diff --git a/resnet50_image_embedding.py b/resnet50_image_embedding.py index 66f3bdb..8e9a7cf 100644 --- 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)]):