update ddcolor with new repo structure (#2634)

CVS-160262
openvinotoolkit · Jan 8, 2025 · ce01b33 · ce01b33
1 parent 513d3bc
commit ce01b33
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/notebooks/ddcolor-image-colorization/ddcolor-image-colorization.ipynb b/notebooks/ddcolor-image-colorization/ddcolor-image-colorization.ipynb
@@ -117,10 +117,16 @@
    },
    "outputs": [],
    "source": [
-    "try:\n",
-    "    from inference.colorization_pipeline_hf import DDColorHF, ImageColorizationPipelineHF\n",
-    "except Exception:\n",
-    "    from inference.colorization_pipeline_hf import DDColorHF, ImageColorizationPipelineHF"
+    "if Path(\"DDColor/inference\").exists():\n",
+    "    try:\n",
+    "        from inference.colorization_pipeline_hf import DDColorHF, ImageColorizationPipelineHF\n",
+    "    except Exception:\n",
+    "        from inference.colorization_pipeline_hf import DDColorHF, ImageColorizationPipelineHF\n",
+    "else:\n",
+    "    try:\n",
+    "        from infer_hf import DDColorHF, ImageColorizationPipelineHF\n",
+    "    except Exception:\n",
+    "        from infer_hf import DDColorHF, ImageColorizationPipelineHF"
    ]
   },
   {