From 582c9748515db9f8d2903deeb175f469bd207a31 Mon Sep 17 00:00:00 2001 From: fakerdaker <52134018+Fakerdaker@users.noreply.github.com> Date: Thu, 25 Jul 2019 03:29:31 -0400 Subject: [PATCH] Colab choose random preview (#316) --- 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