AbdalrhmanRi commited on
Commit
ee72580
·
verified ·
1 Parent(s): 1eb8025

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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': 'cuda'}) # Can use CPU if you want
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: