Update app.py

app.py

@@ -6,12 +6,14 @@ MODEL_NAME = "pujithapsx/address-crossencoder-bge-reranker-v2-m3-finetuned"
 
 print("Loading model...")
 #model = CrossEncoder(MODEL_NAME)
-model = CrossEncoder(
-    MODEL_NAME,
-    trust_remote_code=True,                          # allow custom model code
-    automodel_args={"ignore_mismatched_sizes": True}, # skip size mismatch errors
-    config_args={"model_type": "bert"},              # inject missing model_type
-)
+# model = CrossEncoder(
+#     MODEL_NAME,
+#     trust_remote_code=True,                          # allow custom model code
+#     automodel_args={"ignore_mismatched_sizes": True}, # skip size mismatch errors
+#     config_args={"model_type": "bert"},              # inject missing model_type
+# )
+config   = BertConfig.from_pretrained(MODEL_NAME)   # BertConfig doesn't need model_type
+hf_model = AutoModelForSequenceClassification.from_pretrained(MODEL_NAME, config=config)
 print("Model loaded successfully!")
 
 # STATIC THRESHOLD