Correct, model referencing before assignment

demo_gradio.py

@@ -69,7 +69,7 @@ def process_image_once(inputs, enable_mask):
 
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     model = get_model_on_device(device)
-    model.module.return_masks = enable_mask
+
     image = inputs["image"]
     drawn_boxes = inputs["points"]
     image_tensor = torch.tensor(image).to(device)
@@ -85,6 +85,7 @@ def process_image_once(inputs, enable_mask):
     bboxes = bboxes.unsqueeze(0).to(device)
 
     with torch.no_grad():
+        model.module.return_masks = enable_mask
         outputs, _, _, _, masks = model(img, bboxes)
 
     outputs[0]["pred_boxes"] = outputs[0]["pred_boxes"].cpu()