rushankg commited on
Commit
1303ac0
·
1 Parent(s): 56a13a2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,7 +7,7 @@ import numpy as np
7
  cosine_scores = pickle.load(open('cosine_scores.pkl','rb'))
8
  coursedf = pd.read_pickle('course_df.pkl')
9
 
10
- course_title_list = coursedf['title'].to_list()
11
 
12
  def get_random_course():
13
  row=coursedf.sample(1)
@@ -31,4 +31,4 @@ st.title('DiscoverCourses')
31
  selected_course = st.selectbox('Please select a course',course_title_list)
32
 
33
  if st.button('I want more like this!'):
34
- st.write(recommend(np.where(coursedf['title'] == selected_course)[0][0]))
 
7
  cosine_scores = pickle.load(open('cosine_scores.pkl','rb'))
8
  coursedf = pd.read_pickle('course_df.pkl')
9
 
10
+ course_title_list = [i + ": " + j for i, j in zip(coursedf['ref'].to_list(), coursedf['title'].to_list())]
11
 
12
  def get_random_course():
13
  row=coursedf.sample(1)
 
31
  selected_course = st.selectbox('Please select a course',course_title_list)
32
 
33
  if st.button('I want more like this!'):
34
+ st.write(recommend(np.where((coursedf['ref']+": "+coursedf['title']) == selected_course)[0][0]))