22f1000545 commited on
Commit
3e537af
·
2 Parent(s): 22b8dd29a3b848

Fix merge conflicts - keep correct iris classifier

Browse files
Files changed (2) hide show
  1. app.py +5 -1
  2. requirements.txt +3 -3
app.py CHANGED
@@ -1,16 +1,20 @@
 
1
  from fastapi import FastAPI
2
  from sklearn.datasets import load_iris
3
  from sklearn.tree import DecisionTreeClassifier
4
  import numpy as np
5
 
6
  app = FastAPI()
 
 
7
  iris = load_iris()
8
  model = DecisionTreeClassifier(random_state=42)
9
  model.fit(iris.data, iris.target)
10
  class_names = ["setosa", "versicolor", "virginica"]
11
 
12
  @app.get("/health")
13
- async def health(): return {"status": "ok"}
 
14
 
15
  @app.get("/predict")
16
  async def predict(sl: float, sw: float, pl: float, pw: float):
 
1
+ <<<<<<< HEAD
2
  from fastapi import FastAPI
3
  from sklearn.datasets import load_iris
4
  from sklearn.tree import DecisionTreeClassifier
5
  import numpy as np
6
 
7
  app = FastAPI()
8
+
9
+ # Train model at startup
10
  iris = load_iris()
11
  model = DecisionTreeClassifier(random_state=42)
12
  model.fit(iris.data, iris.target)
13
  class_names = ["setosa", "versicolor", "virginica"]
14
 
15
  @app.get("/health")
16
+ async def health():
17
+ return {"status": "ok"}
18
 
19
  @app.get("/predict")
20
  async def predict(sl: float, sw: float, pl: float, pw: float):
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
- fastapi
2
- uvicorn
3
- scikit-learn
4
  numpy
 
1
+ fastapi
2
+ uvicorn
3
+ scikit-learn
4
  numpy