Improved logging.

app.py

@@ -10,7 +10,6 @@ is_colab = util.is_google_colab()
 
 def run_cmd(command):
     try:
-        print(command)
         call(command, shell=True)
     except KeyboardInterrupt:
         print("Process interrupted")

inference.py

@@ -27,8 +27,10 @@ device = torch.device('cuda' if is_cuda() else 'cpu')
 model = arch.RRDB_Net(3, 3, 64, 23, gc=32, upscale=4, norm_type=None, act_type='leakyrelu', mode='CNA', res_scale=1, upsample_mode='upconv')
 
 if is_cuda():
+    print("Using GPU 🥶")
     model.load_state_dict(torch.load(model_path), strict=True)
 else:
+    print("Using CPU 😒")
     model.load_state_dict(torch.load(model_path, map_location=torch.device('cpu')), strict=True)
 
 model.eval()

inference_manga_v2.py

@@ -19,8 +19,10 @@ device = torch.device('cuda' if is_cuda() else 'cpu')
 model = arch.RRDB_Net(1, 1, 64, 23, gc=32, upscale=4, norm_type=None, act_type='leakyrelu', mode='CNA', res_scale=1, upsample_mode='upconv')
 
 if is_cuda():
+    print("Using GPU 🥶")
     model.load_state_dict(torch.load(model_path), strict=True)
 else:
+    print("Using CPU 😒")
     model.load_state_dict(torch.load(model_path, map_location=torch.device('cpu')), strict=True)
 
 model.eval()