update app.py

app.py

@@ -230,6 +230,9 @@ device = "cuda"
 device = get_device()
 print(f'device={device}')
 
+# make dir
+os.makedirs(output_dir, exist_ok=True)
+
 # initialize groundingdino model
 logger.info(f"initialize groundingdino model...")
 groundingdino_model = load_model_hf(config_file, ckpt_repo_id, ckpt_filenmae)
@@ -326,13 +329,14 @@ def run_grounded_sam(input_image, text_prompt, task_type, inpaint_prompt, box_th
     text_prompt = text_prompt.strip()
     if not ((task_type == 'inpainting' or task_type == 'remove') and mask_source_radio == mask_source_draw):
         if text_prompt == '':
-            return [], gr.Gallery.update(label='Detection prompt is not found!')
+            return [], gr.Gallery.update(label='Detection prompt is not found!😂😂😂😂')
+
+    if input_image is None:
+            return [], gr.Gallery.update(label='Please upload a image!😂😂😂😂')
 
     file_temp = int(time.time())
     logger.info(f'run_grounded_sam_[{file_temp}]_{task_type}_[{text_prompt}]_1_')
 
-    # make dir
-    os.makedirs(output_dir, exist_ok=True)
     # load image
     input_mask_pil = input_image['mask']
     input_mask = np.array(input_mask_pil.convert("L"))  
@@ -362,8 +366,8 @@ def run_grounded_sam(input_image, text_prompt, task_type, inpaint_prompt, box_th
             groundingdino_model, image, text_prompt, box_threshold, text_threshold, device=groundingdino_device
         )
         if boxes_filt.size(0) == 0:
-            logger.info(f'run_grounded_sam_[{file_temp}]_{task_type}_[{text_prompt}]_1_[No objects detected]_')
-            return [], gr.Gallery.update(label='No objects detected, please try others.')
+            logger.info(f'run_grounded_sam_[{file_temp}]_{task_type}_[{text_prompt}]_1_[No objects detected, please try others.]_')
+            return [], gr.Gallery.update(label='No objects detected, please try others.😂😂😂😂')
         boxes_filt_ori = copy.deepcopy(boxes_filt)
 
         pred_dict = {