Linhz commited on
Commit
05712e4
·
verified ·
1 Parent(s): c1e1f38

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ index_loaded = faiss.read_index("sentence_embeddings_index_no_citation.faiss")
26
 
27
  device = 0 if torch.cuda.is_available() else -1
28
  if 'model_embedding' not in st.session_state:
29
- st.session_state.model_embedding = SentenceTransformer('bkai-foundation-models/vietnamese-bi-encoder', device = device)
30
 
31
 
32
 
 
26
 
27
  device = 0 if torch.cuda.is_available() else -1
28
  if 'model_embedding' not in st.session_state:
29
+ st.session_state.model_embedding = SentenceTransformer('bkai-foundation-models/vietnamese-bi-encoder', device = f"cuda:{device}")
30
 
31
 
32