Spaces:

StKirill
/

chatbot

Sleeping

StKirill commited on Feb 18, 2024

Commit

08648ba

verified ·

1 Parent(s): 763c5e6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -408,7 +408,7 @@ def chat_cross_bert(question):
     queries = [question] * len(corpus)
-    tokenized_texts = tokenizer(
         queries, corpus, max_length=MAX_LENGTH, padding=True, truncation=True, return_tensors="pt"
     ).to(device)

     queries = [question] * len(corpus)
+    tokenized_texts = model_cross_encoder.bert_tokenizer(
         queries, corpus, max_length=MAX_LENGTH, padding=True, truncation=True, return_tensors="pt"
     ).to(device)