Update app.py

app.py

@@ -17,8 +17,10 @@ st.write(dbname)
 st.write('Loading Graph to ArangoDB')
 movie_rec_db = load_data_to_ArangoDB(login)
 st.write('Making PyG graph from ArangoDB')
-train_data, val_data, test_data = make_pyg_graph(movie_rec_db)
+data,train_data, val_data, test_data = make_pyg_graph(movie_rec_db)
 st.write('training graph')
-train(train_data, val_data, test_data)
-
+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)
 #---------------------