update prompt in state if only prompt parameter changed

app.py

@@ -113,6 +113,10 @@ def setup_model(loaded_model_setup, prompt, model, seed, num_iterations, enable_
         if new_args_dict == prev_args_dict:
             # If the arguments (excluding prompt) are the same, reuse the loaded setup
             print(f"Arguments (excluding prompt) are the same, reusing loaded setup for {model} model.")
+            
+            # Update the prompt in the loaded_model_setup
+            loaded_model_setup[0].prompt = prompt
+            
             return f"{model} model already loaded with the same configuration.", loaded_model_setup
 
     # If other args differ, proceed with the setup