sanbatte commited on
Commit
a3c4535
·
1 Parent(s): f22a214

agrega mensaje en root

Browse files
Files changed (3) hide show
  1. app/routes/home.py +8 -0
  2. app/routes/prediction.py +0 -1
  3. main.py +9 -4
app/routes/home.py ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ from fastapi import APIRouter
2
+
3
+ router = APIRouter()
4
+
5
+
6
+ @router.get("/")
7
+ def read_root():
8
+ return {"message": "Xepelin MLOps Challenge por Santiago Battezzati"}
app/routes/prediction.py CHANGED
@@ -16,7 +16,6 @@ def predict(request: PredictionRequest):
16
 
17
  prediction_data = df.loc[request.invoiceId]
18
  predictions = model.predict(prediction_data)
19
-
20
  response_data = [
21
  {"invoiceId": invoice_id, "prediction": float(prediction)}
22
  for invoice_id, prediction in zip(request.invoiceId, predictions)
 
16
 
17
  prediction_data = df.loc[request.invoiceId]
18
  predictions = model.predict(prediction_data)
 
19
  response_data = [
20
  {"invoiceId": invoice_id, "prediction": float(prediction)}
21
  for invoice_id, prediction in zip(request.invoiceId, predictions)
main.py CHANGED
@@ -1,11 +1,16 @@
1
- from fastapi import FastAPI, Depends
2
  from app.routes import prediction
3
- from app.models.prediction_models import PredictionRequest, PredictionResponse
4
-
5
 
6
  app = FastAPI()
7
 
8
- # Include three routers
 
 
 
 
 
 
9
  app.include_router(
10
  prediction.router,
11
  prefix="/predict",
 
1
+ from fastapi import FastAPI
2
  from app.routes import prediction
3
+ from app.routes import home
 
4
 
5
  app = FastAPI()
6
 
7
+
8
+ app.include_router(
9
+ home.router,
10
+ tags=["home"],
11
+ )
12
+
13
+
14
  app.include_router(
15
  prediction.router,
16
  prefix="/predict",