Spaces:

SiraH
/

DQA-Llama2-4bit

Sleeping

SiraH commited on Oct 16, 2023

Commit

11debf7

1 Parent(s): 833996b

change model embddding

Files changed (1) hide show

app.py CHANGED Viewed

@@ -197,7 +197,7 @@ def set_custom_prompt():
 @st.cache_resource
 def load_embeddings():
-    embeddings = HuggingFaceEmbeddings(model_name = "all-MiniLM-L6-v2",
                                        model_kwargs = {'device': 'cpu'})
     return embeddings

 @st.cache_resource
 def load_embeddings():
+    embeddings = HuggingFaceEmbeddings(model_name = "thenlper/gte-base",
                                        model_kwargs = {'device': 'cpu'})
     return embeddings