pujithapsx commited on
Commit
9fc5fea
·
verified ·
1 Parent(s): a8cdbd8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -5,7 +5,13 @@ from sentence_transformers import CrossEncoder
5
  MODEL_NAME = "pujithapsx/address-crossencoder-bge-reranker-v2-m3-finetuned"
6
 
7
  print("Loading model...")
8
- model = CrossEncoder(MODEL_NAME)
 
 
 
 
 
 
9
  print("Model loaded successfully!")
10
 
11
  # STATIC THRESHOLD
 
5
  MODEL_NAME = "pujithapsx/address-crossencoder-bge-reranker-v2-m3-finetuned"
6
 
7
  print("Loading model...")
8
+ #model = CrossEncoder(MODEL_NAME)
9
+ model = CrossEncoder(
10
+ MODEL_NAME,
11
+ trust_remote_code=True, # allow custom model code
12
+ automodel_args={"ignore_mismatched_sizes": True}, # skip size mismatch errors
13
+ config_args={"model_type": "bert"}, # inject missing model_type
14
+ )
15
  print("Model loaded successfully!")
16
 
17
  # STATIC THRESHOLD