passed the models folder to training func

app.py

@@ -78,7 +78,7 @@ def online_inference(nlp_models):
             st.write("No named entities found in the text.")
     
 # Streamlit UI for Model Training
-def model_training():
+def model_training(saved_models_dir):
     
     st.title("Model Training")
     
@@ -331,7 +331,7 @@ def main():
     if page == "Online Inference":
         online_inference(nlp_models)
     elif page == "Model Training":
-        model_training()
+        model_training(saved_models_dir)
     elif page == "GEN AI":
         gen_ai()