diff --git a/embedding.png b/embedding.png
deleted file mode 100644
index b154b3a..0000000
Binary files a/embedding.png and /dev/null differ
diff --git a/otk_interactive/otk.py b/otk_interactive/otk.py
index 573ab62..91022af 100644
--- a/otk_interactive/otk.py
+++ b/otk_interactive/otk.py
@@ -15,8 +15,8 @@ def interactive_embeddings(X, y, dims=2, embedding_type="tsne"):
     Returns:
     - fig: Plotly Figure object.
     """
-    embeddings = get_embeddings(
-        X, y, embedding_dims=dims, embedding_type=embedding_type, return_plot=False
+    embeddings, _ = get_embeddings(
+        X, y, embedding_dims=dims, embedding_type=embedding_type, return_plot=True
     )
     df = {f"dim{i+1}": embeddings[:, i] for i in range(dims)}
     df["label"] = y
diff --git a/poetry.lock b/poetry.lock
index 816f023..c09995d 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -1444,22 +1444,28 @@ type = ["mypy (>=1.14.1)"]
 
 [[package]]
 name = "plotly"
-version = "6.0.1"
+version = "6.3.0"
 description = "An open-source interactive data visualization library for Python"
 optional = false
 python-versions = ">=3.8"
 groups = ["main"]
 files = [
-    {file = "plotly-6.0.1-py3-none-any.whl", hash = "sha256:4714db20fea57a435692c548a4eb4fae454f7daddf15f8d8ba7e1045681d7768"},
-    {file = "plotly-6.0.1.tar.gz", hash = "sha256:dd8400229872b6e3c964b099be699f8d00c489a974f2cfccfad5e8240873366b"},
+    {file = "plotly-6.3.0-py3-none-any.whl", hash = "sha256:7ad806edce9d3cdd882eaebaf97c0c9e252043ed1ed3d382c3e3520ec07806d4"},
+    {file = "plotly-6.3.0.tar.gz", hash = "sha256:8840a184d18ccae0f9189c2b9a2943923fd5cae7717b723f36eef78f444e5a73"},
 ]
 
 [package.dependencies]
 narwhals = ">=1.15.1"
+numpy = {version = "*", optional = true, markers = "extra == \"express\""}
 packaging = "*"
 
 [package.extras]
+dev = ["plotly[dev-optional]"]
+dev-build = ["build", "jupyter", "plotly[dev-core]"]
+dev-core = ["pytest", "requests", "ruff (==0.11.12)"]
+dev-optional = ["anywidget", "colorcet", "fiona (<=1.9.6) ; python_version <= \"3.8\"", "geopandas", "inflect", "numpy", "orjson", "pandas", "pdfrw", "pillow", "plotly-geo", "plotly[dev-build]", "plotly[kaleido]", "polars[timezone]", "pyarrow", "pyshp", "pytz", "scikit-image", "scipy", "shapely", "statsmodels", "vaex ; python_version <= \"3.9\"", "xarray"]
 express = ["numpy"]
+kaleido = ["kaleido (>=1.0.0)"]
 
 [[package]]
 name = "pluggy"
@@ -2239,4 +2245,4 @@ files = [
 [metadata]
 lock-version = "2.1"
 python-versions = ">=3.10,<3.12"
-content-hash = "c4cb50309b909475b17a09f49eb0f7ce47f9300c14fc87ed8a486ba3751b3359"
+content-hash = "a30a026b36f68bcd1d88c9840bdc3fae62813414f0d0679d593d4b29cf1fc13a"
diff --git a/pyproject.toml b/pyproject.toml
index 3249bd3..5671f2b 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -10,7 +10,6 @@ license = { text = "MIT License" }
 
 [tool.poetry.dependencies]
 python = ">=3.10,<3.12"
-plotly = ">=6.0.1,<7.0.0"
 optical-toolkit = ">=1.3.0,<2.0.0"
 pandas = ">=2.2.3,<3.0.0"
 tensorflow-io-gcs-filesystem = [
@@ -18,6 +17,7 @@ tensorflow-io-gcs-filesystem = [
     { version = "0.31.0", markers = "sys_platform == 'win32'" }
 ]
 tensorflow = "2.19.0"
+plotly = {extras = ["express"], version = "^6.3.0"}
 
 [tool.poetry.group.dev.dependencies]
 black = "^25.1.0"