From 5a209fb1af23b3c593e72a8967a0b87e7aaf6622 Mon Sep 17 00:00:00 2001
From: fakerdaker <52134018+Fakerdaker@users.noreply.github.com>
Date: Mon, 22 Jul 2019 22:23:57 -0400
Subject: [PATCH] Colab choose random preview

---
 models/ModelBase.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/models/ModelBase.py b/models/ModelBase.py
index 0ff3154..b823a74 100644
--- a/models/ModelBase.py
+++ b/models/ModelBase.py
@@ -102,6 +102,8 @@ class ModelBase(object):
 
         if (self.iter == 0 or ask_override) and self.options['write_preview_history'] and io.is_support_windows():
             choose_preview_history = io.input_bool("Choose image for the preview history? (y/n skip:%s) : " % (yn_str[False]) , False)
+        elif (self.iter == 0 or ask_override) and self.options['write_preview_history'] and io.is_colab():
+            choose_preview_history = io.input_bool("Randomly choose new image for preview history? (y/n ?:help skip:%s) : " % (yn_str[False]), False, help_message="Preview image history will stay stuck with old faces if you reuse the same model on different celebs. Choose no unless you are changing src/dst to a new person")
         else:
             choose_preview_history = False