Update app.py

app.py

@@ -4,6 +4,8 @@ import json
 
 from gradio_client import Client, handle_file
 backend = Client(os.getenv("BACKEND"), hf_token=os.getenv("TOKEN"))
+JS_FUNC1 = os.getenv("JS_FUNC1")
+JS_FUNC2 = os.getenv("JS_FUNC2")
 
 def detect(image):
     result_text = backend.predict(
@@ -73,6 +75,7 @@ with gr.Blocks(css=custom_css) as demo:
                     lbl_aigen.render()
                 with gr.Column():
                     lbl_deepfake.render()
+            gr.HTML("<div id='limit'></div>")
     gr.HTML(MARKDOWN3)
     with gr.Row():
         with gr.Column(scale=1):
@@ -80,6 +83,8 @@ with gr.Blocks(css=custom_css) as demo:
         with gr.Column(scale=5):
             html = gr.HTML()
 
+    demo.load(None, inputs=None, outputs=html, js=JS_FUNC1)
+    limit_button.click(None, js=JS_FUNC2)
     detect_button.click(detect, inputs=[image], outputs=[lbl_overall, lbl_aigen, lbl_deepfake], api_name=False)
 
 demo.queue(api_open=False, default_concurrency_limit=8).launch(server_name="0.0.0.0", show_api=False)