Vamshiboss8055 commited on
Commit
67bc5ef
·
verified ·
1 Parent(s): d4ab75a

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +30 -0
app.py ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from fastapi import FastAPI
2
+ from pydantic import BaseModel
3
+ import joblib
4
+ import numpy as np
5
+
6
+ app = FastAPI(title="Iris Classification API")
7
+
8
+ model = joblib.load("model.joblib")
9
+ labels = ["setosa", "versicolor", "virginica"]
10
+
11
+ class IrisInput(BaseModel):
12
+ sepal_length: float
13
+ sepal_width: float
14
+ petal_length: float
15
+ petal_width: float
16
+
17
+ @app.get("/")
18
+ def home():
19
+ return {"status": "Iris API is running"}
20
+
21
+ @app.post("/predict")
22
+ def predict(data: IrisInput):
23
+ X = np.array([[data.sepal_length, data.sepal_width,
24
+ data.petal_length, data.petal_width]])
25
+ probs = model.predict_proba(X)[0]
26
+ idx = probs.argmax()
27
+ return {
28
+ "prediction": labels[idx],
29
+ "confidence": float(probs[idx])
30
+ }