gamza commited on
Commit
8de1fde
ยท
1 Parent(s): e3724eb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -10,6 +10,10 @@ examples = [["์š”์ฆ˜ ์ž ์ด ์•ˆ ์˜จ๋‹ค"]]
10
 
11
  model = SentenceTransformer('jhgan/ko-sroberta-multitask')
12
 
 
 
 
 
13
  def recommend(message):
14
  embedding = model.encode(message)
15
  df_emb['๊ฑฐ๋ฆฌ'] = df_emb['์„œํ‰์ž„๋ฒ ๋”ฉ'].map(lambda x: cosine_similarity([embedding], [x]).squeeze())
 
10
 
11
  model = SentenceTransformer('jhgan/ko-sroberta-multitask')
12
 
13
+ df = pd.read_pickle('BookData_emb.pkl')
14
+ df_emb = df[['์„œํ‰์ž„๋ฒ ๋”ฉ']].copy()
15
+
16
+
17
  def recommend(message):
18
  embedding = model.encode(message)
19
  df_emb['๊ฑฐ๋ฆฌ'] = df_emb['์„œํ‰์ž„๋ฒ ๋”ฉ'].map(lambda x: cosine_similarity([embedding], [x]).squeeze())