From d8af790fa6f0d943fd453c79d27964ae30c4f8cc Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Fri, 27 Jan 2023 21:54:46 -0500
Subject: [PATCH] Small colab notebook fix.

---
 notebooks/comfyui_colab.ipynb | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/notebooks/comfyui_colab.ipynb b/notebooks/comfyui_colab.ipynb
index 4762c03c..cdf182b8 100644
--- a/notebooks/comfyui_colab.ipynb
+++ b/notebooks/comfyui_colab.ipynb
@@ -106,8 +106,10 @@
         "        break\n",
         "      sock.close()\n",
         "  from google.colab import output\n",
-        "  x = output.serve_kernel_port_as_window(port)\n",
-        "  x = output.serve_kernel_port_as_iframe(port, height=1024)\n",
+        "  output.serve_kernel_port_as_iframe(port, height=1024)\n",
+        "  print(\"to open it in a window you can open this link here:\")\n",
+        "  output.serve_kernel_port_as_window(port)\n",
+        "\n",
         "threading.Thread(target=iframe_thread, daemon=True, args=(8188,)).start()\n",
         "\n",
         "!python main.py"