“Update”

colab.ipynb

@@ -20,10 +20,11 @@
     "from huggingface_hub import InferenceClient\n",
     "from IPython.display import display, clear_output\n",
     "from src.img_gen_colab import generate_image, save_image\n",
-    "from config.config_colab import models, prompts, api_token  # Import from config folder\n",
+    "from config.config_colab import prompts, api_token  # Import from config folder\n",
     "from PIL import Image\n",
     "from google.colab import userdata\n",
     "from datetime import datetime\n",
+    "from config.models import models\n",
     "\n",
     "# Initialize the InferenceClient with the default model\n",
     "client = InferenceClient(models[0][\"name\"], token=api_token)\n",