Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -766,7 +766,7 @@ def respond_pkb(
|
|
766 |
|
767 |
|
768 |
if type(model) == CrossEncoder:
|
769 |
-
sentence_pairs = [[message, v[
|
770 |
scores = model.predict(sentence_pairs, convert_to_tensor=True)
|
771 |
# weights = [1,1]
|
772 |
else:
|
|
|
766 |
|
767 |
|
768 |
if type(model) == CrossEncoder:
|
769 |
+
sentence_pairs = [[message, v[6:]] for v in codes]
|
770 |
scores = model.predict(sentence_pairs, convert_to_tensor=True)
|
771 |
# weights = [1,1]
|
772 |
else:
|