Andre commited on
Commit
df2c30c
·
1 Parent(s): 465c4f7
Files changed (1) hide show
  1. colab.ipynb +2 -0
colab.ipynb CHANGED
@@ -13,6 +13,8 @@
13
  "from img_gen_logic import generate_image\n",
14
  "from config import models, prompts\n",
15
  "\n",
 
 
16
  "\n",
17
  "# Dropdown menu for model selection\n",
18
  "model_dropdown = widgets.Dropdown(\n",
 
13
  "from img_gen_logic import generate_image\n",
14
  "from config import models, prompts\n",
15
  "\n",
16
+ "# Retrieve the Hugging Face token from Colab secrets\n",
17
+ "api_token = userdata.get(\"HF_TOKEN\")\n",
18
  "\n",
19
  "# Dropdown menu for model selection\n",
20
  "model_dropdown = widgets.Dropdown(\n",