Update app.py

Fix "Text" feature name

app.py

@@ -83,22 +83,16 @@ def predict(features: TextInput):
     Fait une prédiction sur un texte donné en utilisant le modèle chargé.
 
     Args:
-        input_data (TextInput): Objet contenant le texte à prédire.
+        features (TextInput): Objet contenant le texte à prédire.
 
     Returns:
-        dict: Dictionnaire contenant la prédiction (0 les articles avec un biais, 1 pour les articles faux, et 2 pour les articles fiable).
-
-    Raises:
-        HTTPException: Si une erreur survient lors de la prédiction.
+        dict: Dictionnaire contenant la prédiction.
     """
     try:
-        # Préparation des données pour la prédiction
-        df = pd.DataFrame({"text": [features.text]})
+        df = pd.DataFrame({"Text": [features.text]})
 
-        # Convertis en tableau 2D (1 échantillon, 1 feature)
-        text_array = np.array(df["text"].tolist()).reshape(1, -1)
+        prediction = model.predict(df)[0]
 
-        prediction = model.predict(text_array)[0]
         return {"prediction": int(prediction)}
 
     except Exception as e: