From 6822fc1b5e3822b72bce90212f54da42fc6a55bb Mon Sep 17 00:00:00 2001
From: fakerdaker <52134018+Fakerdaker@users.noreply.github.com>
Date: Mon, 22 Jul 2019 18:34:12 -0400
Subject: [PATCH] Update model previews in Colab

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

diff --git a/models/ModelBase.py b/models/ModelBase.py
index 0ff3154..e890975 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