Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -28,7 +28,7 @@ def initialize_session_state():
|
|
28 |
st.session_state['vector_store'] = None
|
29 |
if 'embeddings' not in st.session_state:
|
30 |
st.session_state['embeddings'] = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2",
|
31 |
-
model_kwargs={'device': '
|
32 |
if 'translation_states' not in st.session_state:
|
33 |
st.session_state['translation_states'] = {}
|
34 |
if 'message_ids' not in st.session_state:
|
|
|
28 |
st.session_state['vector_store'] = None
|
29 |
if 'embeddings' not in st.session_state:
|
30 |
st.session_state['embeddings'] = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2",
|
31 |
+
model_kwargs={'device': 'cpu'}) # Can use CUDA if you want on your device
|
32 |
if 'translation_states' not in st.session_state:
|
33 |
st.session_state['translation_states'] = {}
|
34 |
if 'message_ids' not in st.session_state:
|