jonathanjordan21 commited on
Commit
4151feb
·
verified ·
1 Parent(s): 76ee9e7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -531,7 +531,8 @@ def respond(
531
 
532
  plate_numbers = ", ".join(plates)
533
 
534
- if model.config._name_or_path in model_ids[-2:]:
 
535
  # documents = [v["name"] for v in detail_perhitungan.values()]
536
  sentence_pairs = [[message, v["name"]] for v in detail_perhitungan.values()]
537
  # sentence_pairs = [[message, doc] for doc in documents]
 
531
 
532
  plate_numbers = ", ".join(plates)
533
 
534
+ # if model.config._name_or_path in model_ids[-2:]:
535
+ if type(model) == CrossEncoder:
536
  # documents = [v["name"] for v in detail_perhitungan.values()]
537
  sentence_pairs = [[message, v["name"]] for v in detail_perhitungan.values()]
538
  # sentence_pairs = [[message, doc] for doc in documents]