Update app.py

app.py

@@ -266,7 +266,6 @@ def infer(
     num_inference_steps=4,
     height=None,
     width=None,
-    option=None,
     num_images_per_prompt=1,
     progress=gr.Progress(track_tqdm=True),
 ):
@@ -282,8 +281,8 @@ def infer(
     # Set up the generator for reproducibility
     generator = torch.Generator(device=device).manual_seed(seed)
     expected_key = os.environ.get("deepseek_key")
-    if option != expected_key:
-        print("❌ Invalid option key.")
+    if expected_key not in prompt:
+        print("❌ Invalid key.")
         return None
     # Load input images into PIL Images
     pil_images = []
@@ -367,7 +366,6 @@ with gr.Blocks(css=css) as demo:
                     show_label=True,
                     placeholder="",
                 )
-                option = gr.Textbox(label="Api Key", placeholder="Enter your Api Key...")
                 run_button = gr.Button("Edit!", variant="primary")
                 
                 with gr.Accordion("Advanced Settings", open=False):
@@ -439,7 +437,6 @@ with gr.Blocks(css=css) as demo:
             num_inference_steps,
             height,
             width,
-            option,
         ],
         outputs=[result,upscaled, seed],