Fix for hf spaces

README.md

@@ -4,7 +4,7 @@ emoji: 💯
 colorFrom: yellow
 colorTo: red
 sdk: streamlit
-app_file: src/visualization/visualize.py
+app_file: app.py
 pinned: false
 ---
 

app.py

@@ -0,0 +1,23 @@
+import streamlit as st
+from src.models.predict_model import predict_model
+
+
+def visualize():
+    st.write("# Summarization  UI")
+    st.markdown(
+        """
+        *For additional questions and inquiries, please contact **Gagan Bhatia** via [LinkedIn](
+        https://www.linkedin.com/in/gbhatia30/) or [Github](https://github.com/gagan3012).*
+        """
+    )
+
+    text = st.text_area("Enter text here")
+    if st.button("Generate Summary"):
+        with st.spinner("Connecting the Dots..."):
+            sumtext = predict_model(text=text)
+        st.write("# Generated Summary:")
+        st.write("{}".format(sumtext))
+
+
+if __name__ == "__main__":
+    visualize()

src/visualization/visualize.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from ..models.predict_model import predict_model
+from src.models.predict_model import predict_model
 
 
 def visualize():