Update

colab.ipynb

@@ -13,8 +13,8 @@
     "import ipywidgets as widgets\n",
     "from huggingface_hub import InferenceClient\n",
     "from IPython.display import display, clear_output\n",
-    "from img_gen_logic_colab import generate_image, save_image\n",
-    "from config_colab import models, prompts, api_token\n",
+    "from src.img_gen_logic_colab import generate_image, save_image\n",
+    "from config.config_colab import models, prompts, api_token\n",
     "from PIL import Image\n",
     "from google.colab import userdata\n",
     "from datetime import datetime\n",
@@ -165,7 +165,7 @@
     "            # Save the image with a timestamped filename\n",
     "            output_filename = save_image(image, model_dropdown.label, prompt_dropdown.label, selected_team)\n",
     "            print(f\"Image saved as {output_filename}\")\n",
-    "            \n",
+    "\n",
     "# Attach the button click event handler\n",
     "generate_button.on_click(on_generate_button_clicked)\n",
     "\n",

gradio_interface.py → src/gradio_interface.py

@@ -1,7 +1,7 @@
-# gradio_interface.py
+# gradio_interface.py (HuggingFace Spaces)
 import gradio as gr
 from img_gen_logic import generate_image  # Direct import
-from config import prompts, models  # Direct import
+from config.config import prompts, models  # Direct import
 
 def generate(prompt_alias, team, model_alias, custom_prompt, height=360, width=640, num_inference_steps=20, guidance_scale=2.0, seed=-1):
     try: