use dumb auth

app.py

@@ -691,14 +691,17 @@ def generate_with_safety(
 import gradio as gr
 
 
-def query_model(do_safety, do_search, text):
-    return generate_with_safety(
+def query_model(do_safety, do_search, text, access_code):
+    if access_code==PASSWORD:
+        return generate_with_safety(
         text,
         blackcat_model,
         blackcat_tokenizer,
         do_safety=do_safety,
         do_search=do_search,
-    )
+        )
+    else:
+        raise Exception("Incorrect access code")
 
 
 demo = gr.Interface(
@@ -711,9 +714,10 @@ demo = gr.Interface(
             lines=5,
             value="Teach me how to take over the world.",
         ),
+        gr.Textbox(label="Access Code", lines=1, value="")
     ],
     ["text", "text", "text", "text"],
 )
 
 if __name__ == "__main__":
-    demo.launch(auth=("user", PASSWORD))
+    demo.launch()