samarthagarwal23 commited on
Commit
15ecd97
1 Parent(s): d9e030f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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) &