File size: 411 Bytes
740d030
 
423ab0d
740d030
 
 
 
 
 
423ab0d
740d030
3e142f4
01bda99
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
from fastapi import FastAPI
from retrain import run_retraining
import requests

app = FastAPI()

@app.post("/retrain")
def retrain():
    result = run_retraining()
    requests.post("https://gdleds-api-fraud.hf.space/reload-model", timeout=120)
    return result

@app.get("/")
def home():
    return {"message" : "Bienvenue sur l'API de retrain du modèle XGBoost pour la prédiction de fraude de transaction"}