Clean app.py

app.py

@@ -10,10 +10,10 @@ from loguru import logger
 import subprocess
 import copy
 import time
-
 import warnings
 
 import torch
+from torchvision.ops import box_convert
 warnings.filterwarnings("ignore")
 
 # grounding DINO
@@ -23,15 +23,11 @@ from groundingdino.util.utils import clean_state_dict
 from groundingdino.util.inference import annotate, load_image, predict
 import groundingdino.datasets.transforms as T
 
-from torchvision.ops import box_convert
-
 # segment anything
 from segment_anything import build_sam, SamPredictor 
 
 from huggingface_hub import hf_hub_download
 
-os.environ["CUDA_VISIBLE_DEVICES"] = "0"
-
 if not os.path.exists('./sam_vit_h_4b8939.pth'):
     logger.info(f"get sam_vit_h_4b8939.pth...")
     result = subprocess.run(['wget', 'https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth'], check=True)
@@ -230,12 +226,11 @@ def grounding_sam(input_image, text_prompt, task_type, box_threshold, text_thres
 
         return output_images, gr.Gallery.update(label='result images')  
 
-    
 groundingDino_model = load_model_hf(config_file, ckpt_repo_id, ckpt_filename, groundingdino_device)
 sam_predictor = SamPredictor(build_sam(checkpoint=sam_checkpoint))
 
 if __name__ == "__main__":
-
+    
     parser = argparse.ArgumentParser("Grounding SAM demo", add_help=True)
     parser.add_argument("--debug", action="store_true", help="using debug mode")
     parser.add_argument("--share", action="store_true", help="share the app")
@@ -278,4 +273,4 @@ if __name__ == "__main__":
         run_button.click(fn=grounding_sam, inputs=[
                         input_image, text_prompt, task_type, box_threshold, text_threshold, iou_threshold], outputs=[gallery, gallery])
 
-    block.launch(share=False, show_api=False, show_error=True)
+    block.launch(debug=args.debug, share=args.share, show_api=False, show_error=True)