Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -31,7 +31,11 @@ def load_embedding_model():
|
|
| 31 |
|
| 32 |
embedding_model = load_embedding_model()
|
| 33 |
|
| 34 |
-
# --- 3.
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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:
|