Update app.py

app.py

@@ -3,16 +3,17 @@ import warnings
 warnings.filterwarnings('ignore')
 
 import subprocess, io, os, sys, time
-os.system("pip install gradio==3.38.0")
+# os.system("pip install gradio==3.38.0")
 import gradio as gr
 
 from loguru import logger
 
+
 os.environ["CUDA_VISIBLE_DEVICES"] = "0"
 
-if os.environ.get('IS_MY_DEBUG') is None:
-    result = subprocess.run(['pip', 'install', '-e', 'GroundingDINO'], check=True)
-    print(f'pip install GroundingDINO = {result}')
+#if os.environ.get('IS_MY_DEBUG') is None:
+#    result = subprocess.run(['pip', 'install', '-e', 'GroundingDINO'], check=True)
+#    print(f'pip install GroundingDINO = {result}')
 
 # result = subprocess.run(['pip', 'list'], check=True)
 # print(f'pip list = {result}')
@@ -63,6 +64,8 @@ from lama_cleaner.helper import (
     resize_max_size,
 )
 
+SECRET_TOKEN = os.getenv('SECRET_TOKEN', 'default_secret')
+
 config_file = 'GroundingDINO/groundingdino/config/GroundingDINO_SwinT_OGC.py'
 ckpt_repo_id = "ShilongLiu/GroundingDINO"
 ckpt_filenmae = "groundingdino_swint_ogc.pth"
@@ -604,12 +607,21 @@ if __name__ == "__main__":
     load_lama_cleaner_model()
     load_ram_model()
 
-    os.system("pip list")
+    # os.system("pip list")
 
     block = gr.Blocks().queue()
     with block:
+        gr.HTML("""
+            <div style="z-index: 100; position: fixed; top: 0px; right: 0px; left: 0px; bottom: 0px; width: 100%; height: 100%; background: white; display: flex; align-items: center; justify-content: center; color: black;">
+            <div style="text-align: center; color: black;">
+            <p style="color: black;">This space is a REST API to programmatically segment an image.</p>
+            <p style="color: black;">Interested in using it? Please use the <a href="https://huggingface.co/spaces/yizhangliu/Grounded-Segment-Anything" target="_blank">original space</a>, thank you!</p>
+            </div>
+            </div>""")
         with gr.Row():
             with gr.Column():
+                text_prompt = gr.Textbox()                           
+                
                 input_image = gr.Image(source='upload', elem_id="image_upload", tool='sketch', type='pil', label="Upload")    
 
                 text_prompt = gr.Textbox(label="Detection Prompt[To detect multiple objects, seperating each name with '.', like this: cat . dog . chair ]", placeholder="Cannot be empty")                                                
@@ -629,7 +641,23 @@ if __name__ == "__main__":
                 image_gallery = gr.Gallery(label="result images", show_label=True, elem_id="gallery", visible=True
                     ).style(preview=True, columns=[5], object_fit="scale-down", height="auto")          
 
-            run_button.click(fn=run_anything_task, inputs=[
-                            input_image, text_prompt, box_threshold, text_threshold, iou_threshold], outputs=[gr.Textbox(), image_gallery, image_gallery], show_progress=True, queue=True)
-
-    block.queue(max_size=1).launch(server_name='0.0.0.0', debug=args.debug, share=args.share)
+            run_button.click(
+                fn=run_anything_task,
+                inputs=[
+                    secret_token,
+                    input_image,
+                    text_prompt,
+                    box_threshold,
+                    text_threshold,
+                    iou_threshold
+                ],
+                outputs=[
+                    gr.Textbox(),
+                    image_gallery,
+                    image_gallery
+                ],
+                show_progress=False,
+                queue=True
+            )
+
+    block.queue(max_size=20).launch(server_name='0.0.0.0', debug=args.debug, share=args.share)