or

1 files changed, 4 insertions, 0 deletions

diff --git a/cli/app/search/live.py b/cli/app/search/live.py
index d98af2f..4d8ad1b 100644
--- a/cli/app/search/live.py
+++ b/cli/app/search/live.py
@@ -479,6 +479,10 @@ class Listener:
       out_img = vs.data2pil(gen_images[0])
       to_pil_time_total += time.time() - to_pil_time
 
+      if out_img is None:
+        print("Got None instead of an image...?")
+        return
+
       save_time = time.time()
       # out_img.save(os.path.join(path_out, "frame_{:05d}.png".format(i)), format='png', compression_level=3)
       out_img.save(pipe.stdin, format='png', compression_level=3)