Upload folder using huggingface_hub

app/routers/predict.py

@@ -44,14 +44,16 @@ async def predict_single(request: PredictionRequest):
         result = pipeline.predict(df)
         
         prediction = result["predictions"][0]
-        probability = result.get("probabilities")[0] if result.get("probabilities") else None
-        
+        proba_row = result.get("probabilities")[0] if result.get("probabilities") else None
+        # proba_row is a list [p_class0, p_class1, ...]; take the positive-class prob for logging
+        proba_scalar = float(proba_row[-1]) if proba_row is not None else None
+
         # Log prediction
         prediction_logger.log_prediction(
             input_data=input_dict,
             prediction=int(prediction),
             model_version="v1",
-            metadata={"probability": float(probability) if probability else None}
+            metadata={"probability": proba_scalar}
         )
         
         # Update metrics
@@ -60,7 +62,7 @@ async def predict_single(request: PredictionRequest):
         
         return PredictionResponse(
             prediction=prediction,
-            probability=probability
+            probability=proba_row
         )
     except Exception as e:
         prediction_counter.labels(model_version="v1", status="error").inc()