Spaces:
Sleeping
Sleeping
File size: 575 Bytes
78fc4f9 53577a8 6b1d291 78fc4f9 6b1d291 53577a8 6b1d291 c7aef65 c8da0d8 c7aef65 78fc4f9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
from fastapi import FastAPI
from train import train_model
from predict import predict
from validate import validate_input
from test import test_api
from pydantic import BaseModel
from validate import TransactionData # Shared schema
app = FastAPI()
@app.get("/")
def root():
return {"message": "LogReg FastAPI is up"}
# Expose APIs
app.add_api_route("/train", train_model, methods=["POST"])
app.add_api_route("/predict", predict, methods=["POST"])
app.add_api_route("/validate", validate_input, methods=["POST"])
app.add_api_route("/test", test_api, methods=["GET"])
|