Spaces:
Runtime error
Runtime error
Commit
·
15ecd97
1
Parent(s):
d9e030f
Update app.py
Browse files
app.py
CHANGED
@@ -2,6 +2,7 @@ import gradio as gr
|
|
2 |
import pandas as pd
|
3 |
import pickle
|
4 |
from sentence_transformers import SentenceTransformer, util
|
|
|
5 |
|
6 |
mdl_name = 'sentence-transformers/all-distilroberta-v1'
|
7 |
model = SentenceTransformer(mdl_name)
|
@@ -12,6 +13,7 @@ with open(embedding_cache_path, "rb") as fIn:
|
|
12 |
|
13 |
embedding_table = cache_data["embeddings"]
|
14 |
reviews = cache_data["data"]
|
|
|
15 |
|
16 |
def user_query_recommend(query, min_p, max_p):
|
17 |
# Embed user query
|
@@ -23,7 +25,6 @@ def user_query_recommend(query, min_p, max_p):
|
|
23 |
|
24 |
# Recommend
|
25 |
recommendations = reviews.copy()
|
26 |
-
recommendations['price'] =recommendations.price.apply(lambda x: re.findall("\d+", x.replace(",","").replace(".00","").replace("$",""))[0]).astype('int')
|
27 |
recommendations['sim'] = sim_scores.T
|
28 |
recommendations = recommendations.sort_values('sim', ascending=False)
|
29 |
recommendations = recommendations.loc[(recommendations.price >= min_p) &
|
|
|
2 |
import pandas as pd
|
3 |
import pickle
|
4 |
from sentence_transformers import SentenceTransformer, util
|
5 |
+
import re
|
6 |
|
7 |
mdl_name = 'sentence-transformers/all-distilroberta-v1'
|
8 |
model = SentenceTransformer(mdl_name)
|
|
|
13 |
|
14 |
embedding_table = cache_data["embeddings"]
|
15 |
reviews = cache_data["data"]
|
16 |
+
reviews['price'] = reviews.price.apply(lambda x: re.findall("\d+", x.replace(",","").replace(".00","").replace("$",""))[0]).astype('int')
|
17 |
|
18 |
def user_query_recommend(query, min_p, max_p):
|
19 |
# Embed user query
|
|
|
25 |
|
26 |
# Recommend
|
27 |
recommendations = reviews.copy()
|
|
|
28 |
recommendations['sim'] = sim_scores.T
|
29 |
recommendations = recommendations.sort_values('sim', ascending=False)
|
30 |
recommendations = recommendations.loc[(recommendations.price >= min_p) &
|