diff --git a/pyproject.toml b/pyproject.toml
index 6411acb..d6cd070 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -55,6 +55,11 @@ spaces = ["gradio>=5.0,<6.0"]
 [project.scripts]
 obliteratus = "obliteratus.cli:main"
 
+[tool.setuptools]
+# The local UI launcher imports the Hugging Face Spaces application module.
+# Include that top-level module in wheels as well as editable/source installs.
+py-modules = ["app"]
+
 [tool.setuptools.packages.find]
 include = ["obliteratus*"]