fix import in samples

openvinotoolkit · Jun 11, 2024 · b9d80b8 · b9d80b8
1 parent 661a5cf
commit b9d80b8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/run_classification.py b/examples/run_classification.py
@@ -284,7 +284,7 @@ def insert_xai(args):
     model = ov.Core().read_model(args.model_path)
 
     # insert XAI branch
-    model_xai = openvino_xai.api.api.insert_xai(
+    model_xai = openvino_xai.insert_xai(
         model,
         task=xai.Task.CLASSIFICATION,
     )
@@ -315,7 +315,7 @@ def insert_xai_w_params(args):
     )
 
     # insert XAI branch
-    model_xai = openvino_xai.api.api.insert_xai(
+    model_xai = openvino_xai.insert_xai(
         model,
         task=xai.Task.CLASSIFICATION,
         insertion_parameters=insertion_parameters,