Update inference_seg.py

inference_seg.py

@@ -24,7 +24,15 @@ def load_model(use_box=False):
 
 @torch.no_grad()
 def run(model, img_path, box=None, device="cpu"):
-    output = model(img_path, box=box)
+    model.move_to_device(device)
+    model.eval()
+    with torch.no_grad():
+        if box is not None:
+            use_box = True
+        else:
+            use_box = False
+        model.use_box = use_box
+        output = model(img_path, box=box)
     mask = output
     return mask
 # import os