Update API service

Convert text into array

app.py

@@ -1,6 +1,7 @@
 import os
 import uvicorn
 import pandas as pd
+import numpy as np
 from pydantic import BaseModel
 from fastapi import FastAPI, HTTPException, status, File, UploadFile
 import mlflow
@@ -94,8 +95,10 @@ def predict(features: TextInput):
         # Préparation des données pour la prédiction
         df = pd.DataFrame({"text": [features.text]})
 
-        # Prédiction
-        prediction = model.predict(df["text"].tolist())[0]
+        # Convertis en tableau 2D (1 échantillon, 1 feature)
+        text_array = np.array(df["text"].tolist()).reshape(1, -1)
+
+        prediction = model.predict(text_array)[0]
         return {"prediction": int(prediction)}
 
     except Exception as e:

requirements.txt

@@ -8,6 +8,7 @@ fastapi
 uvicorn[standard]
 pydantic
 pandas
+numpy
 gunicorn
 openpyxl
 boto3