Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -50,7 +50,7 @@ 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 |
|
| 52 |
#The bi-encoder will retrieve 100 documents. We use a cross-encoder, to re-rank the results list to improve the quality
|
| 53 |
-
cross_encoder = CrossEncoder(option2,
|
| 54 |
|
| 55 |
passages = []
|
| 56 |
|
|
|
|
| 50 |
top_k = 32 #Number of passages we want to retrieve with the bi-encoder
|
| 51 |
|
| 52 |
#The bi-encoder will retrieve 100 documents. We use a cross-encoder, to re-rank the results list to improve the quality
|
| 53 |
+
cross_encoder = CrossEncoder(option2, device='cpu')
|
| 54 |
|
| 55 |
passages = []
|
| 56 |
|