“Update”

app.py

@@ -1,18 +1,6 @@
-# app.py ANDRE
-import sys
-import os
-
-
-# Add the src folder to the Python path
-# Solves all problems w subfolders - option2
-src_path = os.path.abspath(os.path.join(os.path.dirname(__file__), "src"))
-if src_path not in sys.path:
-    sys.path.append(src_path)
-
+# app.py 
 # Import gradio_interface
 from gradio_interface import demo
 
-
-if __name__ == "__main__":
-    # Launch the Gradio app
-    demo.launch()
+# Launch the Gradio app
+demo.launch()

colab.ipynb

@@ -19,7 +19,7 @@
     "import ipywidgets as widgets\n",
     "from huggingface_hub import InferenceClient\n",
     "from IPython.display import display, clear_output\n",
-    "from src.img_gen_logic_colab import generate_image, save_image\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 PIL import Image\n",
     "from google.colab import userdata\n",

config/config.py

@@ -6,13 +6,6 @@ from config.models import models
 # Retrieve the Hugging Face token
 api_token = os.getenv("HF_TOKEN")
 
-# Debugging: Check if the Hugging Face token is available
-if not api_token:
-    print("ERROR: Hugging Face token (HF_TOKEN) is missing. Please set it as an environment variable or in Colab secrets.")
-else:
-    print("Hugging Face token loaded successfully.")
-
-
 # Debugging: Print prompt and model options
 print("Prompt Options:", [p["alias"] for p in prompts])
 print("Model Options:", [m["alias"] for m in models])