Update app.py

app.py

@@ -475,8 +475,12 @@ def concatenate_images_vertical(image1, image2):
 mask_source_draw = "draw a mask on input image"
 mask_source_segment = "type what to detect below"
 
-def run_anything_task(input_image, text_prompt, box_threshold, text_threshold, 
+def run_anything_task(secret_token, input_image, text_prompt, box_threshold, text_threshold, 
             iou_threshold, cleaner_size_limit=1080):
+    if secret_token != SECRET_TOKEN:
+        raise gr.Error(
+            f'Invalid secret token. Please fork the original space if you want to use it for yourself.')
+
     task_type = "segment"             
 
     text_prompt = text_prompt.strip()
@@ -620,7 +624,8 @@ if __name__ == "__main__":
             </div>""")
         with gr.Row():
             with gr.Column():
-                text_prompt = gr.Textbox()                           
+                secret_token = gr.Textbox()
+                text_prompt = gr.Textbox()
                 
                 input_image = gr.Image(source='upload', elem_id="image_upload", tool='sketch', type='pil', label="Upload")