Spaces:

vishwask
/

witp_poc

Runtime error

vishwask commited on Sep 20, 2023

Commit

ddbca10

1 Parent(s): 57d037d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -69,7 +69,7 @@ def load_data():
 @st.cache_resource
-def load_model(docs):
     #embeddings = HuggingFaceInstructEmbeddings(model_name="hkunlp/instructor-large",model_kwargs={"device":DEVICE})
     embeddings = HuggingFaceInstructEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2",model_kwargs={"device":DEVICE})
     print(DEVICE)

 @st.cache_resource
+def load_model(_docs):
     #embeddings = HuggingFaceInstructEmbeddings(model_name="hkunlp/instructor-large",model_kwargs={"device":DEVICE})
     embeddings = HuggingFaceInstructEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2",model_kwargs={"device":DEVICE})
     print(DEVICE)