Update face_deid_ct.py

face_deid_ct.py

@@ -4,9 +4,23 @@ import numpy as np
 import cv2
 from matplotlib import pyplot as plt
 import random
-from tqdm.notebook import tqdm
 import time
+import tqdm
+from IPython.core.display import display, HTML
+
+# Determine if we are in a Jupyter notebook
+try:
+    shell = get_ipython().__class__.__name__
+    if shell == 'ZMQInteractiveShell':
+        # We are in Jupyter, use tqdm.notebook
+        from tqdm.notebook import tqdm
+    else:
+        raise Exception()
+except:
+    # We are in a terminal, use standard tqdm
+    from tqdm import tqdm
 
+    
 FACE_MAX_VALUE = 50
 FACE_MIN_VALUE = -125