correction

app.py

@@ -29,6 +29,7 @@ global isModelDefined
 isModelDefined = False
 
 def defineModel():
+    """ Manging the control flow of this to support ZeroGPU behavior"""
     global model
     global isModelDefined
     global cuda
@@ -397,6 +398,7 @@ def train_model_persona(positive_text, negative_text):
         positive_list,
         negative_list,
         output_suffixes)
+    defineModel()
     global model
     model.reset()
     output_model = ControlVector.train(model, tokenizer, dataset)
@@ -423,6 +425,7 @@ def train_model_facts(positive_text, negative_text):
         negative_text,
         fact_suffixes
     )
+    defineModel()
     global model
     model.reset()
     output_model = ControlVector.train(model, tokenizer, dataset)