Add login

app.py

@@ -1,6 +1,7 @@
 import csv
 import json
 import math
+import os
 import secrets
 from pathlib import Path
 from typing import cast
@@ -17,6 +18,13 @@ from PIL import Image, ImageFilter, ImageOps
 
 DEVICE = "cuda"
 
+USER = os.getenv("USER")
+PASSWORD = os.getenv("PASSWORD")
+
+if not USER or not PASSWORD:
+    msg = "USER and PASSWORD must be set"
+    raise ValueError(msg)
+
 MAX_SEED = np.iinfo(np.int32).max
 
 SYSTEM_PROMPT = r"""This two-panel split-frame image showcases a furniture in as a product shot versus styled in a room.
@@ -613,4 +621,4 @@ with gr.Blocks(css=css) as demo:
         preprocess=False,
     )
 
-demo.launch()
+demo.launch(auth=[(USER, PASSWORD)])