diff --git a/resnet_image_embedding.py b/resnet_image_embedding.py
index d10f224..1c0a8d5 100644
--- a/resnet_image_embedding.py
+++ b/resnet_image_embedding.py
@@ -26,7 +26,7 @@ class ResnetImageEmbedding(Operator):
     """
     PyTorch model for image embedding.
     """
-    def __init__(self, model_name: str, framework: str = 'pytorch') -> None:
+    def __init__(self, model_name: str, framework: str = 'pytorch', weights_path: str = None) -> None:
         super().__init__()
         sys.path.append(str(Path(__file__).parent))
         if framework == 'pytorch':