Update app.py

app.py

@@ -1,10 +1,29 @@
 import streamlit as st
-from logic import *
+from logic2 import *
+
+st.title('PyG - Movie Recommendation')
+st.image('Image_graph.png', caption='Bi-Partite Graph')
+st.plotly_chart(fig, use_container_width=True)
+
+# load Hgraph
+data = load_hetero_data()
+
+# Load Model
+model = load_model(data)
+
+
+
+
+
+
+
+
+
 
-st.title('ArangoDB - PyG')
 
 
 #---------------------
+'''
 st.write('Making Graph')
 global movie_mappings, user_mapping, ratings_df, m_id, id_map, sampled_md
 movie_mappings, user_mapping, ratings_df, m_id, id_map, sampled_md = make_graph()
@@ -24,4 +43,6 @@ model = load_model(train_data, val_data, test_data)
 user_id = st.number_input('Insert a number')
 ans = get_recommendation(model,data,user_id)
 st.write(ans)
+'''
 #---------------------
+