diff --git a/inference.ipynb b/inference.ipynb
index 2ba6992..937c3b8 100644
--- a/inference.ipynb
+++ b/inference.ipynb
@@ -123,7 +123,9 @@
         "\n",
         "\n",
         "print(\" * Running UI...\")\n",
-        "!python app.py --checkpoint_path ./checkpoints/ --port 7865 --device_id 0 --share true --bf16 {bf16}"
+        "\n",
+        "!pip install e .",
+        "!acestep --checkpoint_path ./checkpoints/ --port 7865 --device_id 0 --share true --bf16 {bf16}"
       ],
       "metadata": {
         "cellView": "form",
@@ -133,4 +135,4 @@
       "outputs": []
     }
   ]
-}
\ No newline at end of file
+}