Add option to trust remote code

src/app.py

@@ -14,7 +14,10 @@ async def lifespan(app: FastAPI) -> AsyncIterator[None]:
     """Setup and teardown events of the app"""
     # Setup
     config = Config()
-    app.state.model = SimilarityClassifierModel(model_name=config.reranker_model_name)
+    app.state.model = SimilarityClassifierModel(
+        model_name=config.reranker_model_name,
+        trust_remote_code=config.trust_remote_code,
+    )
     app.state.api_key = config.api_key
     yield
     # Teardown

src/config.py

@@ -6,6 +6,7 @@
 class Config(BaseSettings):
     reranker_model_name: str = "BAAI/bge-reranker-v2-m3"
     api_key: str | None = None
+    trust_remote_code: bool = False
 
 
 def get_log_config(level: str = "INFO"):

src/inference.py

@@ -10,10 +10,16 @@
 
 
 class SimilarityClassifierModel:
-    def __init__(self, model_name: str) -> None:
-        self.tokenizer = AutoTokenizer.from_pretrained(model_name, cache_dir=CACHE_DIR)
+    def __init__(self, model_name: str, trust_remote_code: bool = False) -> None:
+        self.tokenizer = AutoTokenizer.from_pretrained(
+            model_name,
+            cache_dir=CACHE_DIR,
+            trust_remote_code=trust_remote_code,
+        )
         self.model = AutoModelForSequenceClassification.from_pretrained(
-            model_name, cache_dir=CACHE_DIR
+            model_name,
+            cache_dir=CACHE_DIR,
+            trust_remote_code=trust_remote_code,
         )
 
         self.device = "cuda" if torch.cuda.is_available() else "cpu"