Update app.py
Browse files
app.py
CHANGED
@@ -69,7 +69,7 @@ def load_data():
|
|
69 |
|
70 |
|
71 |
@st.cache_resource
|
72 |
-
def load_model(
|
73 |
#embeddings = HuggingFaceInstructEmbeddings(model_name="hkunlp/instructor-large",model_kwargs={"device":DEVICE})
|
74 |
embeddings = HuggingFaceInstructEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2",model_kwargs={"device":DEVICE})
|
75 |
print(DEVICE)
|
|
|
69 |
|
70 |
|
71 |
@st.cache_resource
|
72 |
+
def load_model(_docs):
|
73 |
#embeddings = HuggingFaceInstructEmbeddings(model_name="hkunlp/instructor-large",model_kwargs={"device":DEVICE})
|
74 |
embeddings = HuggingFaceInstructEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2",model_kwargs={"device":DEVICE})
|
75 |
print(DEVICE)
|