krisha06 commited on
Commit
fc83c21
·
verified ·
1 Parent(s): 5ad41a2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -31,7 +31,11 @@ def load_embedding_model():
31
 
32
  embedding_model = load_embedding_model()
33
 
34
- # --- 3. Initialize ChromaDB ---
 
 
 
 
35
  chroma_client = chromadb.PersistentClient(path="./chroma_db") # Removed the telemetry argument
36
 
37
  @st.cache_resource
@@ -57,10 +61,6 @@ def load_or_generate_recipe_embeddings(recipes_df):
57
 
58
  collection = load_or_generate_recipe_embeddings(recipes_df)
59
 
60
- # --- 4. Generate & Store Embeddings ---
61
- def get_sentence_transformer_embeddings(text):
62
- return embedding_model.encode(text).tolist()
63
-
64
  # --- 5. Retrieve Similar Recipes ---
65
  def retrieve_recipes(query, top_k=3):
66
  try:
 
31
 
32
  embedding_model = load_embedding_model()
33
 
34
+ # --- 3. Generate & Store Embeddings ---
35
+ def get_sentence_transformer_embeddings(text):
36
+ return embedding_model.encode(text).tolist()
37
+
38
+ # --- 4. Initialize ChromaDB ---
39
  chroma_client = chromadb.PersistentClient(path="./chroma_db") # Removed the telemetry argument
40
 
41
  @st.cache_resource
 
61
 
62
  collection = load_or_generate_recipe_embeddings(recipes_df)
63
 
 
 
 
 
64
  # --- 5. Retrieve Similar Recipes ---
65
  def retrieve_recipes(query, top_k=3):
66
  try: