Spaces:

anushka3306
/

iris-fastapi

Sleeping

Update app.py

ba97d96 verified 6 months ago

648 Bytes

from fastapi import FastAPI
from sklearn.datasets import load_iris
from sklearn.tree import DecisionTreeClassifier
import numpy as np

app = FastAPI()

# Train model at startup
iris = load_iris()
model = DecisionTreeClassifier(max_depth=2, random_state=42)
model.fit(iris.data, iris.target)
class_names = ["setosa", "versicolor", "virginica"]

@app.get("/health")
async def health():
    return {"status": "ok"}

@app.get("/predict")
async def predict(sl: float, sw: float, pl: float, pw: float):
    features = np.array([[sl, sw, pl, pw]])
    pred = int(model.predict(features)[0])
    return {"prediction": pred, "class_name": class_names[pred]}

	from fastapi import FastAPI
	from sklearn.datasets import load_iris
	from sklearn.tree import DecisionTreeClassifier
	import numpy as np

	app = FastAPI()

	# Train model at startup
	iris = load_iris()
	model = DecisionTreeClassifier(max_depth=2, random_state=42)
	model.fit(iris.data, iris.target)
	class_names = ["setosa", "versicolor", "virginica"]

	@app.get("/health")
	async def health():
	return {"status": "ok"}

	@app.get("/predict")
	async def predict(sl: float, sw: float, pl: float, pw: float):
	features = np.array([[sl, sw, pl, pw]])
	pred = int(model.predict(features)[0])
	return {"prediction": pred, "class_name": class_names[pred]}