bug fix

models/common.py

@@ -15,7 +15,6 @@ import pandas as pd
 import requests
 import torch
 import torch.nn as nn
-from IPython.display import display
 from PIL import Image
 from torch.cuda import amp
 
@@ -529,8 +528,7 @@ class Detections:
                 s += '(no detections)'
 
             im = Image.fromarray(im.astype(np.uint8)) if isinstance(im, np.ndarray) else im  # from np
-            if show:
-                display(im) if is_notebook() else im.show(self.files[i])
+            
             if save:
                 f = self.files[i]
                 im.save(save_dir / f)  # save