Update

app2.py

@@ -1,12 +1,5 @@
 # main.py
-import sys
-import os
-
-# Add the /src directory to the Python path
-sys.path.append(os.path.join(os.path.dirname(__file__), 'src'))
-
-# Import necessary modules
-from src.gradio_interface import demo
+from src.gradio_interface import demo  # Direct import
 
 if __name__ == "__main__":
     # Launch the Gradio app

src/gradio_interface.py

@@ -1,7 +1,7 @@
 # src/gradio_interface.py
 import gradio as gr
-from src.img_gen_logic import generate_image
-from src.config import prompts, models
+from img_gen_logic import generate_image  # Direct import
+from 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:

src/img_gen_logic

@@ -3,7 +3,7 @@ import random
 from huggingface_hub import InferenceClient
 from PIL import Image
 from datetime import datetime
-from src.config import api_token, models, prompts
+from config import api_token, models, prompts  # Direct import
 
 def generate_image(prompt_alias, team, model_alias, custom_prompt, height=360, width=640, num_inference_steps=20, guidance_scale=2.0, seed=-1):
     # Debugging: Check if the token is available