Spaces:
Build error
Build error
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 |
|