Update base/app.py

base/app.py

@@ -10,6 +10,10 @@ import torchvision
 import random
 from huggingface_hub import snapshot_download
 
+# Login function for authentication
+def custom_auth(username, password):
+    return password == "aitutor"
+
 config_path = "./base/configs/sample.yaml"
 args = OmegaConf.load("./base/configs/sample.yaml")
 device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -74,7 +78,7 @@ title = """
 """
 
 with gr.Blocks(css='style.css') as demo:
-    gr.Markdown("<font color=red size=10><center>LaVie: Text-to-Video generation</center></font>")
+    gr.Markdown("<font color=red size=10><center>Pixio Text-to-Video generation</center></font>")
     with gr.Column():
         with gr.Row(elem_id="col-container"):
             # inputs = [prompt, seed_inp, ddim_steps]
@@ -121,6 +125,4 @@ with gr.Blocks(css='style.css') as demo:
     submit_btn.click(infer, inputs, outputs)
     # share_button.click(None, [], [], _js=share_js)
 
-demo.queue(max_size=12).launch()
-
-
+demo.queue(max_size=18).launch(auth=custom_auth)