Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -45,7 +45,7 @@ st.sidebar.success("Load Successfully!")
|
|
45 |
# print("Warning: No GPU found. Please add GPU to your notebook")
|
46 |
|
47 |
#We use the Bi-Encoder to encode all passages, so that we can use it with sematic search
|
48 |
-
bi_encoder = SentenceTransformer(option1,
|
49 |
bi_encoder.max_seq_length = 256 #Truncate long passages to 256 tokens
|
50 |
top_k = 32 #Number of passages we want to retrieve with the bi-encoder
|
51 |
|
|
|
45 |
# print("Warning: No GPU found. Please add GPU to your notebook")
|
46 |
|
47 |
#We use the Bi-Encoder to encode all passages, so that we can use it with sematic search
|
48 |
+
bi_encoder = SentenceTransformer(option1,device='cpu')
|
49 |
bi_encoder.max_seq_length = 256 #Truncate long passages to 256 tokens
|
50 |
top_k = 32 #Number of passages we want to retrieve with the bi-encoder
|
51 |
|