Spaces:
Paused
Paused
Update app.py
Browse files
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 |
|