StKirill commited on
Commit
fab4024
·
verified ·
1 Parent(s): 7c4ae65

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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()