diff --git a/mainscripts/Trainer.py b/mainscripts/Trainer.py
index 97a790c..cf4bc97 100644
--- a/mainscripts/Trainer.py
+++ b/mainscripts/Trainer.py
@@ -171,9 +171,12 @@ def trainerThread (s2c, c2s, e,
                             model_save()
                             is_reached_goal = True
                             io.log_info ('You can use preview now.')
-                            if os.path.exists("/home/deepfake/interact_dict.pkl"):
-                                model_save()
-                                os._exit(1)
+                        if os.path.exists(answer_filename):
+                            with open(answer_filename, 'rb') as file:
+                                parameters = pickle.load(file)
+                                if parameters['NoInteractiveMode']:
+                                        model_save()
+                                        os._exit(1)
                 
                 need_save = False
                 while time.time() - last_save_time >= save_interval_min*60:
diff --git a/models/ModelBase.py b/models/ModelBase.py
index d06bbf9..2a4f755 100644
--- a/models/ModelBase.py
+++ b/models/ModelBase.py
@@ -295,10 +295,13 @@ class ModelBase(object):
                 self.choose_preview_history = io.input_bool("Randomly choose new image for preview history", 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")
 
     def ask_target_iter(self, default_value=0):
-        if os.path.exists("/home/deepfake/interact_dict.pkl"):
-            default_target_iter = self.load_or_def_option('target_iter', default_value + 10000)
-            self.options['target_iter'] = max(0, default_target_iter + 10000)
-            print("TARGET ITERATION: " + str(self.options['target_iter']))
+        if os.path.exists(answer_filename):
+            with open(answer_filename, 'rb') as file:
+               parameters = pickle.load(file)
+               if parameters['NoInteractiveMode']:
+                 default_target_iter = self.load_or_def_option('target_iter', default_value + 10000)
+                 self.options['target_iter'] = max(0, default_target_iter + 10000)
+                 print("TARGET ITERATION: " + str(self.options['target_iter']))
         else:
             default_target_iter = self.load_or_def_option('target_iter', default_value)
             self.options['target_iter'] = max(0, io.input_int("Target iteration", default_target_iter))