Olivier-52 commited on
Commit
362c37d
·
1 Parent(s): 91f0a33

Add Xgboost lib

Browse files
Files changed (2) hide show
  1. app.py +1 -0
  2. requirements.txt +2 -1
app.py CHANGED
@@ -4,6 +4,7 @@ import pandas as pd
4
  from pydantic import BaseModel
5
  from fastapi import FastAPI, File, UploadFile
6
  import mlflow
 
7
  import os
8
  from dotenv import load_dotenv
9
 
 
4
  from pydantic import BaseModel
5
  from fastapi import FastAPI, File, UploadFile
6
  import mlflow
7
+ from xgboost import XGBClassifier
8
  import os
9
  from dotenv import load_dotenv
10
 
requirements.txt CHANGED
@@ -10,4 +10,5 @@ gunicorn
10
  openpyxl
11
  boto3
12
  python-multipart
13
- dotenv
 
 
10
  openpyxl
11
  boto3
12
  python-multipart
13
+ dotenv
14
+ xgboost