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()
|