Update app.py

app.py

@@ -5,8 +5,8 @@ import torch
 import numpy as np
 
 cosine_scores = pickle.load(open('cosine_scores.pkl','rb'))
-coursedf = pd.read_pickle('course_df.pkl')          # course_df uses titles to generate course recommendations
-course_df_new = pd.read_pickle('course_df_new.pkl') #course_df_new makes recommendations using the entire description
+coursedf = pd.read_pickle('course_df_new.pkl')          # course_df uses titles to generate course recommendations
+#course_df_new = pd.read_pickle('course_df_new.pkl') #course_df_new makes recommendations using the entire description
 
 course_title_list = [i + ": " + j for i, j in zip(coursedf['ref'].to_list(), coursedf['title'].to_list())]
 
@@ -42,7 +42,7 @@ if maincol1.button('Recommend by title',use_container_width=True):
         course_id=coursedf.iloc[index,0]
         st.subheader(course_id+": "+result)
         with st.expander("See description"):
-            st.write(course_df_new.iloc[index,3]) #Using the new coursedf because it has proper descriptions for each course
+            st.write(coursedf.iloc[index,3]) #Using the new coursedf because it has proper descriptions for each course
         link = "[ExploreCourses](https://explorecourses.stanford.edu/search?q="+course_id+"+"+result.replace(" ","+")+")"
         st.markdown(link, unsafe_allow_html=True)
         link = "[Carta](https://carta-beta.stanford.edu/results/"+course_id+")"
@@ -51,13 +51,13 @@ if maincol1.button('Recommend by title',use_container_width=True):
         
 if maincol2.button('Recommend by description',use_container_width=True):
     index_new=np.where((coursedf['ref']+": "+coursedf['title']) == selected_course)[0][0]
-    rec_list=course_df_new.iloc[index_new,2]
-    for result in rec_list:
+    rec_list=coursedf.iloc[index_new,2]
+    for result in rec_list[1:]:
         index=np.where(coursedf['title'] == result)[0][0]
         course_id=coursedf.iloc[index,0]
         st.subheader(course_id+": "+result)
         with st.expander("See description"):
-            st.write(course_df_new.iloc[index,3]) #Using the new coursedf because it has proper descriptions for each course
+            st.write(coursedf.iloc[index,3]) #Using the new coursedf because it has proper descriptions for each course
         link = "[ExploreCourses](https://explorecourses.stanford.edu/search?q="+course_id+"+"+result.replace(" ","+")+")"
         st.markdown(link, unsafe_allow_html=True)
         link = "[Carta](https://carta-beta.stanford.edu/results/"+course_id+")"