Update transformers_ensemble with custom model_name support for FinBERTHead

macro_sentiment/transformers_ensemble.py

@@ -67,9 +67,10 @@ class SentimentHead:
 
 
 class FinBERTHead(SentimentHead):
-    def __init__(self, device="cpu"):
+    """FinBERT for financial news sentiment. Accepts custom model_name for fine-tuned variants."""
+    def __init__(self, model_name="ProsusAI/finbert", device="cpu"):
         super().__init__(
-            model_name="ProsusAI/finbert",
+            model_name=model_name,
             label_map={0: "positive", 1: "negative", 2: "neutral"},
             score_map={"positive": 1.0, "negative": -1.0, "neutral": 0.0},
             device=device,
@@ -168,11 +169,11 @@ class TopicRouter:
 
 
 class TransformerEnsemble:
-    def __init__(self, device="cpu", use_router=True):
+    def __init__(self, device="cpu", use_router=True, finbert_model="ProsusAI/finbert"):
         self.device = device
         self.use_router = use_router
         self.heads = {
-            "finbert": FinBERTHead(device),
+            "finbert": FinBERTHead(model_name=finbert_model, device=device),
             "policy": FinancialRoBERTaHead(device),
             "climate": ClimateBERTHead(device),
         }