Update app.py
Browse files
app.py
CHANGED
@@ -302,7 +302,7 @@ def chat_bert_context(question, history):
|
|
302 |
return answer
|
303 |
|
304 |
#-------------------------------------Bi-BERT-Encoder------------------------------------------#
|
305 |
-
|
306 |
# Define function for mean-pooling
|
307 |
def mean_pool(token_embeds: torch.tensor, attention_mask: torch.tensor) -> torch.tensor:
|
308 |
in_mask = attention_mask.unsqueeze(-1).expand(token_embeds.size()).float()
|
|
|
302 |
return answer
|
303 |
|
304 |
#-------------------------------------Bi-BERT-Encoder------------------------------------------#
|
305 |
+
MAX_LENGTH = 128
|
306 |
# Define function for mean-pooling
|
307 |
def mean_pool(token_embeds: torch.tensor, attention_mask: torch.tensor) -> torch.tensor:
|
308 |
in_mask = attention_mask.unsqueeze(-1).expand(token_embeds.size()).float()
|