Spaces:
Sleeping
Sleeping
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 |
|