ensure torch.no_grad()

app.py

@@ -124,6 +124,7 @@ def draw_box(box, draw, label):
         draw.text((box[0], box[1]), label, font=font)
 
 
+@torch.no_grad()
 def inference(raw_image, specified_tags, tagging_model_type, tagging_model, grounding_dino_model, sam_model):
     print(f"Start processing, image size {raw_image.size}")
     raw_image = raw_image.convert("RGB")