Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -532,7 +532,7 @@ def respond(
|
|
532 |
if model.config._name_or_path in model_ids[-2:]:
|
533 |
# documents = [v["name"] for v in detail_perhitungan.values()]
|
534 |
# sentence_pairs = [[message, v["name"]] for v in detail_perhitungan.values()]
|
535 |
-
|
536 |
scores = model.predict(sentence_pairs, convert_to_tensor=True)
|
537 |
weights = [1,1,1,1,1,1,1,1,1]
|
538 |
# scores = [x["score"] for x in model.rank(message, documents)]
|
|
|
532 |
if model.config._name_or_path in model_ids[-2:]:
|
533 |
# documents = [v["name"] for v in detail_perhitungan.values()]
|
534 |
# sentence_pairs = [[message, v["name"]] for v in detail_perhitungan.values()]
|
535 |
+
sentence_pairs = [[message, doc] for doc in documents]
|
536 |
scores = model.predict(sentence_pairs, convert_to_tensor=True)
|
537 |
weights = [1,1,1,1,1,1,1,1,1]
|
538 |
# scores = [x["score"] for x in model.rank(message, documents)]
|