Deal with missing colorama.

diff --git a/cnn-svrt.py b/cnn-svrt.py
index 338e145..0c63b85 100755 (executable)
--- a/cnn-svrt.py
+++ b/cnn-svrt.py
@@ -29,7 +29,10 @@ import distutils.util
 import re
 import signal
 
 import re
 import signal
 

-from colorama import Fore, Back, Style
+try:
+    from colorama import Fore, Back, Style
+except ImportError:
+    Fore, Back, Style = '', '', ''

 
 # Pytorch
 
 
 # Pytorch