From 0af572281863d359e671bee1188608f4ec215a70 Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Fri, 17 Jan 2020 21:16:59 +0100
Subject: or

---
 cli/app/search/live.py | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'cli/app')

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)
-- 
cgit v1.2.3-70-g09d2