test-streamlit / app.py
rushankg's picture
Upload 3 files
17d7d2e
raw
history blame
780 Bytes
import streamlit as st
import pickle
import pandas as pd
cosine_scores = pickle.load(open('cosine_scores.pkl','rb'))
coursedf = pd.DataFrame(pickle.load(open('course_df.pkl','rb')))
course_title_list = courses['title'].to_list()
def get_random_course():
row=courses.sample(1)
return row['ref'], row['title']
def recommend(coursename):
pairs = {}
for i in range(len(coursedf)):
pairs[coursedf.iloc[i,1]]=cosine_scores[index][i]
sorttemp = sorted(pairs.items(), key=lambda x:x[1], reverse=True)
sorted_final = dict(sorttemp[1:31])
return list(sorted_final.keys())
st.set_page_config(page_title='DiscoverCourses', page_icon=':book:')
st.title('DiscoverCourses')
selected_course = st.selectbox('Please select a course',course_title_list)