Update app.py
Browse files
app.py
CHANGED
@@ -369,7 +369,7 @@ def chat_bi_bert(question):
|
|
369 |
|
370 |
|
371 |
#-------------------------------------Bi+Cross-BERT-Encoder------------------------------------------#
|
372 |
-
|
373 |
|
374 |
#Define class for CrossEncoderBert
|
375 |
class CrossEncoderBert(torch.nn.Module):
|
|
|
369 |
|
370 |
|
371 |
#-------------------------------------Bi+Cross-BERT-Encoder------------------------------------------#
|
372 |
+
inverted_answer = dict(enumerate(df.answer.tolist()))
|
373 |
|
374 |
#Define class for CrossEncoderBert
|
375 |
class CrossEncoderBert(torch.nn.Module):
|