correction

app.py

@@ -397,6 +397,7 @@ def train_model_persona(positive_text, negative_text):
         positive_list,
         negative_list,
         output_suffixes)
+    global model
     model.reset()
     output_model = ControlVector.train(model, tokenizer, dataset)
     # Write file to temporary directory returning the path to Gradio for download
@@ -422,7 +423,8 @@ def train_model_facts(positive_text, negative_text):
         negative_text,
         fact_suffixes
     )
-
+    global model
+    model.reset()
     output_model = ControlVector.train(model, tokenizer, dataset)
     filename = re.sub(r'[ <>:"/\\|?*]', '', positive_text) + '_'
     temp_file = tempfile.NamedTemporaryFile(