Update app.py

app.py

@@ -2,6 +2,12 @@ from transformers import pipeline
 import gradio as gr
 import os
 
+def auth(user_name, password):
+    if user_name == os.getenv("USER_NAME") and password == os.getenv("PASSWORD"):
+        return True
+    else:
+        return False
+
 hf_token = os.getenv("HF_TOKEN")
 
 model_repository_name = "Kosuke-Yamada/t5_ad_generation2"
@@ -21,4 +27,4 @@ with gr.Blocks() as demo:
                 output_text = gr.components.Textbox(lines=1, label="生成広告")
 
     submit_button.click(main, inputs=[input_text], outputs=[output_text])
-demo.launch()
+demo.launch(auth=auth)