sarahselim18 commited on
Commit
2f52104
·
1 Parent(s): 8b4af38

changed get sorted cos sim

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -211,8 +211,8 @@ def get_sorted_cosine_similarity(embeddings_metadata):
211
  # TODO: Compute cosine similarity between input sentence and categories
212
  # TODO: Update category embeddings if category not found
213
  ##########################################
214
- cat = categories[index]
215
- cat_embed = category_embeddings[cat]
216
  # Calc cosine sim
217
  cat_scores.append((index, np.dot(input_embedding,cat_embed)))
218
  # Store doc_id and score as a tuple
@@ -342,7 +342,7 @@ st.subheader(
342
  # key="categories"
343
  # )
344
  categories = st.text_input(
345
- label="Categories", key="categories", value="Flowers Colors Cars Weather Food"
346
  )
347
  print(st.session_state["categories"])
348
  print(type(st.session_state["categories"]))
 
211
  # TODO: Compute cosine similarity between input sentence and categories
212
  # TODO: Update category embeddings if category not found
213
  ##########################################
214
+ #cat = categories[index]
215
+ cat_embed = category_embeddings[index]
216
  # Calc cosine sim
217
  cat_scores.append((index, np.dot(input_embedding,cat_embed)))
218
  # Store doc_id and score as a tuple
 
342
  # key="categories"
343
  # )
344
  categories = st.text_input(
345
+ label="Categories", key="categories", value="Flowers Colors Cars Weather Food",
346
  )
347
  print(st.session_state["categories"])
348
  print(type(st.session_state["categories"]))