diff --git a/frontend.py b/frontend.py
index 0c40227..b87e99c 100644
--- a/frontend.py
+++ b/frontend.py
@@ -130,6 +130,7 @@ def main(args):
database.set_image_output_folder(args.image_output_folder)
database.connect(args.db)
+ app.config["TITLE"] = args.title
app.run(host="0.0.0.0", port=args.port)
database.safe_disconnect()
@@ -146,6 +147,11 @@ if __name__ == "__main__":
"--db", type=str, default="happysd.db", help="Path to SQLite database file"
)
+ # Add an argument to set the title of service
+ parser.add_argument(
+ "--title", type=str, default="Happy Diffusion", help="Title of the webpage"
+ )
+
# Add an argument to set the path of the database file
parser.add_argument(
"--image-output-folder",
diff --git a/templates/index.html b/templates/index.html
index 1597c3d..da8d3a8 100644
--- a/templates/index.html
+++ b/templates/index.html
@@ -3,7 +3,7 @@
- Happy Diffusion (Private Access) | 9pm
+ {{ config.TITLE }}