update app.py

app.py

@@ -54,14 +54,6 @@ from io import BytesIO
 from diffusers import StableDiffusionInpaintPipeline
 from huggingface_hub import hf_hub_download
 
-def get_device():
-    from numba import cuda
-    if cuda.is_available():
-        device = 'cuda:0' # cuda.get_current_device()
-    else:
-        device = 'cpu'
-    return device
-
 def load_model_hf(model_config_path, repo_id, filename, device='cpu'):
     args = SLConfig.fromfile(model_config_path) 
     model = build_model(args)
@@ -108,7 +100,11 @@ def plot_boxes_to_image(image_pil, tgt):
             bbox = (x0, y0, w + x0, y0 + h)
         # bbox = draw.textbbox((x0, y0), str(label))
         draw.rectangle(bbox, fill=color)
-        draw.text((x0, y0), str(label), fill="white")
+        font = os.path.join(cv2.__path__[0],'qt','fonts','DejaVuSans.ttf')
+        font_size = 36
+        new_font = ImageFont.truetype(font, font_size)
+
+        draw.text((x0+2, y0+2), str(label), font=new_font, fill="white")
 
         mask_draw.rectangle([x0, y0, x1, y1], fill=255, width=6)
 
@@ -228,9 +224,8 @@ ckpt_repo_id = "ShilongLiu/GroundingDINO"
 ckpt_filenmae = "groundingdino_swint_ogc.pth"
 sam_checkpoint = './sam_vit_h_4b8939.pth' 
 output_dir = "outputs"
-device = "cuda"
+device = evice = 'cuda' if torch.cuda.is_available() else 'cpu'
 
-device = get_device()
 print(f'device={device}')
 
 # make dir
@@ -540,13 +535,13 @@ if __name__ == "__main__":
                 run_button = gr.Button(label="Run")
                 with gr.Accordion("Advanced options", open=False):
                     box_threshold = gr.Slider(
-                        label="Box Threshold", minimum=0.0, maximum=1.0, value=0.6, step=0.001
+                        label="Box Threshold", minimum=0.0, maximum=1.0, value=0.3, step=0.001
                     )
                     text_threshold = gr.Slider(
-                        label="Text Threshold", minimum=0.0, maximum=1.0, value=0.5, step=0.001
+                        label="Text Threshold", minimum=0.0, maximum=1.0, value=0.25, step=0.001
                     )
                     iou_threshold = gr.Slider(
-                        label="IOU Threshold", minimum=0.0, maximum=1.0, value=0.5, step=0.001
+                        label="IOU Threshold", minimum=0.0, maximum=1.0, value=0.8, step=0.001
                     )                    
                     inpaint_mode = gr.Radio(["merge", "first"], value="merge", label="inpaint_mode")
                     with gr.Row():