diff --git a/requirements.txt b/requirements.txt
index 7fa5aa5..183f161 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -109,7 +109,6 @@ tensorboard
 tensorboard-data-server
 tensorboard-plugin-wit
 tensorflow
-tensorflow-docs
 tensorflow-estimator
 termcolor==1.1.0
 terminado==0.12.1
diff --git a/trainer.py b/trainer.py
index ec08195..11a58a7 100644
--- a/trainer.py
+++ b/trainer.py
@@ -88,7 +88,8 @@ def trainSimplaeModel(model, epochs, tran_iterater, test_iterater, model_name, t
     history = model.fit(tran_iterater, validation_data=test_iterater, epochs=epochs, callbacks=[CustomCallback(text_output, progrss_bar, graph)],)
     model_path = os.path.join(st.session_state['output_folder'], model_name)
     st.session_state['model_folder'] =  st.session_state["output_folder"]
-    #st.write(model_path)
+    st.markdown(f'##### Model output : {model_path}')
+    #st.write(f'Output folder is : {model_path}')
     model.save(model_path)
     return history