Update

colab.ipynb

@@ -110,7 +110,6 @@
     "# Output area to display the image\n",
     "output = widgets.Output()\n",
     "\n",
-    "# Function to handle button click event\n",
     "def on_generate_button_clicked(b):\n",
     "    with output:\n",
     "        clear_output(wait=True)  # Clear previous output\n",
@@ -141,7 +140,10 @@
     "\n",
     "        # Generate the image\n",
     "        print(\"=== Debug: Calling generate_image ===\")\n",
-    "        image, message = generate_image(selected_prompt, selected_team, selected_model, height, width, num_inference_steps, guidance_scale, seed, custom_prompt)\n",
+    "        image, message = generate_image(\n",
+    "            selected_prompt, selected_team, selected_model, height, width,\n",
+    "            num_inference_steps, guidance_scale, seed, custom_prompt, api_token\n",
+    "        )\n",
     "\n",
     "        # Debug: Check the output of generate_image\n",
     "        print(\"=== Debug: generate_image Output ===\")\n",