Update app.py

app.py

@@ -50,12 +50,6 @@ if torch.cuda.is_available():
 else:
     pipe = None
     refiner = None
-
-def check_secret_token(token: str) -> str:
-    """Raises an error if the token does not match the secret token."""
-    if token != SECRET_TOKEN:
-        raise ValueError("Invalid secret token!")
-    return token
     
 def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
     if randomize_seed:
@@ -77,7 +71,11 @@ 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:
+             apply_refiner: bool = False,
+             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:
@@ -272,12 +270,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,