torch 2.4 + peejm/finbert (safetensors) + transformers 4.46.3

Dockerfile

@@ -14,7 +14,7 @@ WORKDIR /home/user/app
 
 # CPU-only torch from PyTorch index (skip CUDA wheels, ~200MB vs ~2GB)
 RUN pip install --user --no-cache-dir --index-url https://download.pytorch.org/whl/cpu \
-    torch==2.6.0
+    torch==2.4.1
 
 # App deps — all pinned to known-compatible versions
 RUN pip install --user --no-cache-dir \
@@ -28,7 +28,7 @@ RUN pip install --user --no-cache-dir \
     "einops>=0.7" \
     "safetensors>=0.4" \
     "tqdm>=4.66" \
-    "transformers>=4.40,<5.0" \
+    "transformers==4.46.3" \
     "chronos-forecasting>=1.5.2" \
     "timesfm[torch]>=1.3.0"
 

app.py

@@ -242,7 +242,7 @@ def forecast_timesfm(symbol: str, lookback_days: int = 180, pred_days: int = 30)
 # -----------------------------------------------------------------------------
 # FinBERT — ProsusAI/finbert
 # -----------------------------------------------------------------------------
-FINBERT_MODEL_ID = "ProsusAI/finbert"
+FINBERT_MODEL_ID = "peejm/finbert-financial-sentiment"
 _finbert = None