jonathanjordan21 commited on
Commit
b5fed14
·
verified ·
1 Parent(s): 32aedb1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -766,7 +766,7 @@ def respond_pkb(
766
 
767
 
768
  if type(model) == CrossEncoder:
769
- sentence_pairs = [[message, v["name"]] for v in detail_perhitungan.values()]
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: