jayash391 commited on
Commit
3d713f1
·
verified ·
1 Parent(s): 049d07c

Update sherlock2.py

Browse files
Files changed (1) hide show
  1. 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
- embedding_dict = {"embedding": response["embedding"]} # Create dictionary
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}")