fixed?

app.py

@@ -7,7 +7,6 @@ import zipfile
 import importlib
 import wandb
 from contextlib import redirect_stdout, redirect_stderr
-import spaces
 
 USE_WANDB = "WANDB_API_KEY" in os.environ
 if USE_WANDB:
@@ -795,28 +794,29 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                  to the run log to preserve the live console stream.
                  """
                  try:
-                    # Import or reload the package module so decorators execute on import.
-                    with open(self.log_path, "a", encoding="utf-8") as lf, \
-                            redirect_stdout(lf), redirect_stderr(lf):
+                    # Ensure the project directory is on sys.path so a plain `import auto_forge` finds the local file
+                    script_dir = os.path.dirname(os.path.abspath(__file__))
+                    if script_dir not in sys.path:
+                        sys.path.insert(0, script_dir)
+
+                    # Import the local auto_forge module using a plain import (no importlib utilities)
+                    with open(self.log_path, "a", encoding="utf-8") as lf, redirect_stdout(lf), redirect_stderr(lf):
                         try:
-                            if "autoforge.auto_forge" in sys.modules:
-                                module = importlib.reload(sys.modules["autoforge.auto_forge"])
-                            else:
-                                module = importlib.import_module("autoforge.auto_forge")
+                            # Use plain import; if already imported, the module's top-level code won't re-run.
+                            import auto_forge
 
-                            # Provide CLI-like argv for the parser inside the module
+                            # Provide argv for the module's CLI parsing and call main()
                             sys.argv = ["autoforge"] + (self.cmd[1:] if len(self.cmd) > 1 else [])
-
-                            if hasattr(module, "main"):
+                            if hasattr(auto_forge, "main"):
                                 try:
-                                    module.main()
+                                    auto_forge.main()
                                     self.returncode = 0
                                 except SystemExit as se:
                                     self.returncode = se.code if isinstance(se.code, int) or se.code is None else 0
                             else:
-                                raise AttributeError("autoforge.auto_forge does not expose a main() function")
+                                raise AttributeError("auto_forge module does not expose a main() function")
                         except Exception as e:
-                            lf.write(f"\nERROR while importing/running autoforge.auto_forge: {exc_text(e)}\n")
+                            lf.write(f"\nERROR while importing/running auto_forge: {exc_text(e)}\n")
                             self.exc = e
                             if isinstance(e, SystemExit):
                                 self.returncode = e.code if isinstance(e.code, int) or e.code is None else 1