update app.py

app.py

@@ -13,12 +13,12 @@ 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}')
+    logger.info(f'pip install GroundingDINO = {result}')
 
 logger.info(f"Start app...")
 
 # result = subprocess.run(['pip', 'list'], check=True)
-# print(f'pip list = {result}')
+# logger.info(f'pip list = {result}')
 
 sys.path.insert(0, './GroundingDINO')
 
@@ -132,7 +132,7 @@ def load_model_hf(model_config_path, repo_id, filename, device='cpu'):
     cache_file = hf_hub_download(repo_id=repo_id, filename=filename)
     checkpoint = torch.load(cache_file, map_location=device)
     log = model.load_state_dict(clean_state_dict(checkpoint['model']), strict=False)
-    print("Model loaded from {} \n => {}".format(cache_file, log))
+    logger.info("Model loaded from {} \n => {}".format(cache_file, log))
     _ = model.eval()
     return model    
 
@@ -289,7 +289,7 @@ def set_device(args):
         device = args.cuda if torch.cuda.is_available() else 'cpu'
     else:
         device = 'cpu'
-    print(f'device={device}')
+    logger.info(f'device={device}')
 
 def load_groundingdino_model(device):
     # initialize groundingdino model
@@ -302,7 +302,7 @@ def get_sam_vit_h_4b8939():
     if not os.path.exists('./sam_vit_h_4b8939.pth'):
         logger.info(f"get sam_vit_h_4b8939.pth...")
         result = subprocess.run(['wget', '-nv', 'https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth'], check=True)
-        print(f'wget sam_vit_h_4b8939.pth result = {result}')
+        logger.info(f'wget sam_vit_h_4b8939.pth result = {result}')
 
 def load_sam_model(device):
     # initialize SAM
@@ -1091,10 +1091,11 @@ def main_gradio(args):
                         <a href="https://huggingface.co/spaces/yizhangliu/Grounded-Segment-Anything?duplicate=true"><img style="display: inline; margin-top: 0em; margin-bottom: 0em" src="https://bit.ly/3gLdBN6" alt="Duplicate Space" /></a></p>'
         gr.Markdown(DESCRIPTION)
 
-    print(f'device = {device}')
-    print(f'torch.cuda.is_available = {torch.cuda.is_available()}')
+    logger.info(f'device = {device}')
+    logger.info(f'torch.cuda.is_available = {torch.cuda.is_available()}')
     computer_info()
     block.queue(max_size=10, api_open=False)
+    logger.info(f"Start a gradio server[{os.getpid()}]: http://0.0.0.0:{args.port}")
     block.launch(server_name='0.0.0.0', server_port=args.port, debug=args.debug, share=args.share, show_api=False)
 
 if __name__ == "__main__":
@@ -1104,7 +1105,7 @@ if __name__ == "__main__":
     parser.add_argument("--port", "-p", type=int, default=7860, help="port")
     parser.add_argument("--cuda", "-c", type=str, default='cuda:0', help="cuda")
     args, _ = parser.parse_known_args()
-    print(f'args = {args}')
+    logger.info(f'args = {args}')
 
     if os.environ.get('IS_MY_DEBUG') is None:
         os.system("pip list")