Update app.py

app.py

@@ -81,7 +81,9 @@ def generate(prompt: str,
              guidance_scale_refiner: float = 5.0,
              num_inference_steps_base: int = 50,
              num_inference_steps_refiner: int = 50,
-             apply_refiner: bool = False) -> PIL.Image.Image:
+             secret_token: str = '') -> PIL.Image.Image:
+    if secret_token != SECRET_TOKEN:
+        raise ValueError("Invalid secret token!")
     generator = torch.Generator().manual_seed(seed)
 
     if not use_negative_prompt:
@@ -276,12 +278,6 @@ with gr.Blocks(css='style.css') as demo:
         secret_token,
     ]
     prompt.submit(
-        fn=check_secret_token,
-        inputs=[secret_token],
-        outputs=gr.outputs.Void(),
-        queue=False,
-        api_name=False,
-    ).then(
         fn=randomize_seed_fn,
         inputs=[seed, randomize_seed],
         outputs=seed,