Fix single-row prediction error and enable high-speed HF transfers

Dockerfile

@@ -4,7 +4,8 @@ FROM python:3.11-slim
 RUN useradd -m -u 1000 user
 USER user
 ENV HOME=/home/user \
-    PATH=/home/user/.local/bin:$PATH
+    PATH=/home/user/.local/bin:$PATH \
+    HF_HUB_ENABLE_HF_TRANSFER=1
 
 WORKDIR $HOME/app
 

matrix/benchmark.py

@@ -116,13 +116,25 @@ class _SAPModel:
         return self
 
     def predict(self, X):
-        preds = self._model.predict(X)
+        # sap_rpt_oss fails on single-row prediction due to an internal concatenation bug.
+        # We work around this by doubling the row if len(X) == 1.
+        is_single = len(X) == 1
+        X_input = pd.concat([X, X]) if is_single else X
+        
+        preds = self._model.predict(X_input)
+        if is_single:
+            preds = preds[:1]
+            
         if not self._real and self.task == "classification":
             preds = self._le.inverse_transform(preds)
         return preds
 
     def predict_proba(self, X):
-        return self._model.predict_proba(X)
+        is_single = len(X) == 1
+        X_input = pd.concat([X, X]) if is_single else X
+        
+        proba = self._model.predict_proba(X_input)
+        return proba[:1] if is_single else proba
 
     @property
     def label(self):

requirements.txt

@@ -4,6 +4,8 @@ uvicorn[standard]>=0.29.0
 python-multipart>=0.0.9
 python-dotenv>=1.0.0
 jinja2>=3.1.3
+hf-transfer
+hf_xet
 
 # Data & Science Stack
 numpy==1.26.4