Update

colab.ipynb

@@ -14,6 +14,9 @@
     "from img_gen_logic import generate_image\n",
     "from config import models, prompts\n",
     "\n",
+    "# Initialize the InferenceClient with the default model\n",
+    "client = InferenceClient(models[0][\"name\"], token=api_token)\n",
+    "\n",
     "# Dropdown menu for model selection\n",
     "model_dropdown = widgets.Dropdown(\n",
     "    options=[(model[\"alias\"], model[\"name\"]) for model in models],\n",