“Update”

src/img_gen.py

@@ -1,4 +1,13 @@
 # img_gen.py
+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)
+    
 import random
 from huggingface_hub import InferenceClient
 from datetime import datetime