Skip to content

Commit

Permalink
reformat file
Browse files Browse the repository at this point in the history
  • Loading branch information
ccheng26 committed Apr 25, 2024
1 parent 855d112 commit d1214e1
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion 02-household-queries/chainlit-household-bot.py
Original file line number Diff line number Diff line change
Expand Up @@ -219,7 +219,9 @@ async def set_embedding():
model=model_name, google_api_key=GOOGLE_API_KEY
)
elif embedding in OPEN_SOURCE_EMBEDDINGS:
embedding_function = SentenceTransformerEmbeddings(model_name="all-MiniLM-L6-v2")
embedding_function = SentenceTransformerEmbeddings(
model_name="all-MiniLM-L6-v2"
)
elif embedding in HUGGING_FACE_EMBEDDINGS:
model_name = embedding.split("::")[1]
embedding_function = HuggingFaceEmbeddings(model_name=model_name)
Expand Down

0 comments on commit d1214e1

Please sign in to comment.