araeyn commited on
Commit
44981c3
·
verified ·
1 Parent(s): aad1466

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -46,7 +46,11 @@ encode_kwargs = {'normalize_embeddings': True}
46
  encode_kwargs=encode_kwargs,
47
  show_progress=True,
48
  )"""
49
- embedding = HuggingFaceInferenceAPIEmbeddings(api_key=os.environ["HUGGINGFACE_API_KEY"], model=model_name)
 
 
 
 
50
 
51
  async def echo(websocket):
52
  global retriever, conversational_rag_chain
 
46
  encode_kwargs=encode_kwargs,
47
  show_progress=True,
48
  )"""
49
+ hf_embeddings = HuggingFaceInferenceAPIEmbeddings(
50
+ api_key=os.environ["HUGGINGFACE_API_KEY"],
51
+ model_name="sentence-transformers/all-MiniLM-l6-v2"
52
+ )
53
+
54
 
55
  async def echo(websocket):
56
  global retriever, conversational_rag_chain