Add models and gitignore

.gitignore

@@ -0,0 +1 @@
+.DS_Store

app.py

@@ -2,79 +2,78 @@ import pickle
 import streamlit as st
 import numpy as np
 
-
 st.header("Book Recommender System")
 
-# model = pickle.load(open("artifacts/model.pkl", "rb"))
-# book_names = pickle.load(open("artifacts/book_names.pkl", "rb"))
-# final_ratings = pickle.load(open("artifacts/final_ratings.pkl", "rb"))
-# book_pivot = pickle.load(open("artifacts/book_pivot.pkl", "rb"))
+model = pickle.load(open("artifacts/model.pkl", "rb"))
+book_names = pickle.load(open("artifacts/book_names.pkl", "rb"))
+final_ratings = pickle.load(open("artifacts/final_ratings.pkl", "rb"))
+book_pivot = pickle.load(open("artifacts/book_pivot.pkl", "rb"))
 
-# def fetch_poster(suggestion):
-#     bookNames = []
-#     idsIndex = []
-#     posterUrl = []
+def fetch_poster(suggestion):
+    bookNames = []
+    idsIndex = []
+    posterUrl = []
 
-#     for bookId in suggestion[0]:        
-#         name = book_pivot.index[bookId]
+    for bookId in suggestion[0]:        
+        name = book_pivot.index[bookId]
         
-#         bookNames.append(book_pivot.index[bookId])
+        bookNames.append(book_pivot.index[bookId])
         
-#     for name in bookNames:        
-#         ids = np.where(final_ratings['title'] == name)[0][0]
-#         idsIndex.append(ids)
+    for name in bookNames:        
+        ids = np.where(final_ratings['title'] == name)[0][0]
+        idsIndex.append(ids)
 
-#     for idx in idsIndex:
-#         row = final_ratings.iloc[idx]
-#         url = row['img_url']
-#         posterUrl.append(url)
+    for idx in idsIndex:
+        row = final_ratings.iloc[idx]
+        url = row['img_url']
+        posterUrl.append(url)
 
-#     return posterUrl
+    return posterUrl
 
-# def recommend_book(bookName):
-#     bookList = []
+def recommend_book(bookName):
+    bookList = []
 
-#     book_id = np.where(book_pivot.index == bookName)[0][0]
+    book_id = np.where(book_pivot.index == bookName)[0][0]
     
-#     distance, suggestion = model.kneighbors(book_pivot.iloc[book_id,:].values.reshape(1, -1), n_neighbors=5)
+    distance, suggestion = model.kneighbors(book_pivot.iloc[book_id,:].values.reshape(1, -1), n_neighbors=5)
 
-#     poster_url = fetch_poster(suggestion)
+    poster_url = fetch_poster(suggestion)
 
-#     for i in range(len(suggestion)):
-#         books = book_pivot.index[suggestion[i]]
+    for i in range(len(suggestion)):
+        books = book_pivot.index[suggestion[i]]
 
-#         for j in books:
-#             bookList.append(j)
+        for j in books:
+            bookList.append(j)
 
-#     return bookList, poster_url
+    return bookList, poster_url
 
-# selected_books = st.selectbox(
-#     "Select a book", 
-#     book_names
-# )
+selected_books = st.selectbox(
+    "Select a book", 
+    book_names
+)
 
-# if st.button("Show Recommendations"):
-#     recommendations, posterUrls = recommend_book(selected_books)
+if st.button("Show Recommendations"):
+    recommendations, posterUrls = recommend_book(selected_books)
     
-#     st.subheader("Recommendations")
+    st.subheader("Recommendations")
 
-#     col1, col2, col3, col4 = st.columns(4)
+    col1, col2, col3, col4 = st.columns(4)
 
-#     for url in posterUrls:
-#         print(url)
+    for url in posterUrls:
+        print(url)
 
-#     with col1:
-#         st.text(recommendations[1])
-#         st.image(posterUrls[1])
+    with col1:
+        st.text(recommendations[1])
+        st.image(posterUrls[1])
 
-#     with col2:
-#         st.text(recommendations[2])
-#         st.image(posterUrls[2])
+    with col2:
+        st.text(recommendations[2])
+        st.image(posterUrls[2])
 
-#     with col3:
-#         st.text(recommendations[3])
-#         st.image(posterUrls[3])
+    with col3:
+        st.text(recommendations[3])
+        st.image(posterUrls[3])
 
-#     with col4:
-#         st.text(recommendations[4])
-#         st.image(posterUrls[4])
+    with col4:
+        st.text(recommendations[4])
+        st.image(posterUrls[4])

artifacts/book_names.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:794d08c3eb7ee65bf95008096b99fdcba5376e3794e313ade38d5781451e46a2
+size 20131

artifacts/book_pivot.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a84ce7b17434dc9cf1740e636e9c2a795f5c87f19b7b7e7414b3e8b2da5c97c0
+size 5298853

artifacts/final_ratings.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f47e068289bf4cc38a6240dda847abcb487c5ecd5f8ce1924fc2149692f56ab6
+size 3811459

artifacts/model.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1aa3452bce5d5c6bee719a6aac3490a2cf39c0f417388a31893ffdc0df6f3317
+size 183212