Update app.py

fix pytorch model loading

app.py

@@ -51,14 +51,13 @@ try:
     print("Modèle scikit-learn chargé avec succès.")
 except mlflow.exceptions.MlflowException:
     try:
-        # Si échec, essayer de charger un modèle Transformers
-        model = mlflow.transformers.load_model(model_uri)
-        print("Modèle Transformers chargé avec succès.")
-    except Exception as e:
+        model = mlflow.pytorch.load_model(model_uri)
+        print("Modèle PyTorch chargé avec succès.")
+    except mlflow.exceptions.MlflowException as e:
         raise HTTPException(
             status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
             detail=f"Erreur lors du chargement du modèle : {e}"
-        )
+    )
 
 app = FastAPI(
     title="Climate Fake News Detector API",