Update app.py
Browse files
app.py
CHANGED
@@ -4,13 +4,13 @@ import pickle
|
|
4 |
from sentiment import get_sentiment
|
5 |
|
6 |
# Load the data
|
7 |
-
novel_list = pickle.load(open('
|
8 |
novel_list['english_publisher'] = novel_list['english_publisher'].fillna('unknown')
|
9 |
name_list = novel_list['name'].values
|
10 |
|
11 |
def recommend(novel, slider_start):
|
12 |
try:
|
13 |
-
similarity = pickle.load(open('
|
14 |
novel_index = novel_list[novel_list['name'] == novel].index[0]
|
15 |
distances = similarity[novel_index]
|
16 |
new_novel_list = sorted(list(enumerate(distances)), reverse=True, key=lambda x: x[1])[slider_start:slider_start+9]
|
|
|
4 |
from sentiment import get_sentiment
|
5 |
|
6 |
# Load the data
|
7 |
+
novel_list = pickle.load(open('data/novel_list.pkl', 'rb'))
|
8 |
novel_list['english_publisher'] = novel_list['english_publisher'].fillna('unknown')
|
9 |
name_list = novel_list['name'].values
|
10 |
|
11 |
def recommend(novel, slider_start):
|
12 |
try:
|
13 |
+
similarity = pickle.load(open('data/similarity.pkl', 'rb'))
|
14 |
novel_index = novel_list[novel_list['name'] == novel].index[0]
|
15 |
distances = similarity[novel_index]
|
16 |
new_novel_list = sorted(list(enumerate(distances)), reverse=True, key=lambda x: x[1])[slider_start:slider_start+9]
|