Fixes and almost done

README.md

@@ -78,3 +78,6 @@ Next if we would like to test the model and visualise the results we can run:
 t5s visualize
 ```
 And this would create a streamlit app for testing
+
+
+![carbon (7)](https://user-images.githubusercontent.com/49101362/129279588-17271a4c-7258-4208-a94d-89e5b97b6cd0.png)

src/models/predict_model.py

@@ -14,4 +14,3 @@ def predict_model(text: str):
     model.load_model(model_type=params["model_type"], model_dir=params["model_dir"])
     pre_summary = model.predict(text)
     return pre_summary
-

src/visualization/visualize.py

@@ -1,6 +1,6 @@
 import streamlit as st
 
-from ..models import predict_model
+from src.models.predict_model import predict_model
 
 
 def visualize():
@@ -15,7 +15,7 @@ def visualize():
     text = st.text_area("Enter text here")
     if st.button("Generate Summary"):
         with st.spinner("Connecting the Dots..."):
-            sumtext = predict_model.predict_model(text=text)
+            sumtext = predict_model(text=text)
         st.write("# Generated Summary:")
         st.write("{}".format(sumtext))