Spaces:
Sleeping
Sleeping
Update sherlock2.py
Browse files- sherlock2.py +1 -2
sherlock2.py
CHANGED
@@ -59,8 +59,7 @@ def generate_embeddings_from_documents(extracted_text):
|
|
59 |
# Determine the appropriate task type (e.g., "RETRIEVAL_DOCUMENT" for search/similarity)
|
60 |
task_type = "RETRIEVAL_DOCUMENT"
|
61 |
response = genai.embed_content(model=embed_model, content=text, task_type=task_type)
|
62 |
-
|
63 |
-
embeddings.append(embedding_dict)
|
64 |
return embeddings
|
65 |
except Exception as e:
|
66 |
st.error(f"Error generating embeddings: {e}")
|
|
|
59 |
# Determine the appropriate task type (e.g., "RETRIEVAL_DOCUMENT" for search/similarity)
|
60 |
task_type = "RETRIEVAL_DOCUMENT"
|
61 |
response = genai.embed_content(model=embed_model, content=text, task_type=task_type)
|
62 |
+
embeddings.append(response["embedding"])
|
|
|
63 |
return embeddings
|
64 |
except Exception as e:
|
65 |
st.error(f"Error generating embeddings: {e}")
|