sravan837 commited on
Commit
d4e3479
·
1 Parent(s): 371f086

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +31 -0
app.py ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from fastapi import FastAPI
2
+ import joblib
3
+ import pandas as pd
4
+ from pydantic import BaseModel
5
+
6
+ # Load the trained model
7
+ model = joblib.load("slope_stability_model.pkl")
8
+
9
+ # Initialize FastAPI
10
+ app = FastAPI()
11
+
12
+ # Define request format using Pydantic
13
+ class SlopeStabilityInput(BaseModel):
14
+ unit_weight: float
15
+ cohesion: float
16
+ friction_angle: float
17
+ slope_angle: float
18
+ slope_height: float
19
+ water_pressure_ratio: float
20
+ reinforcement_type: str
21
+
22
+ # Define API endpoint
23
+ @app.post("/predict")
24
+ def predict_slope_stability(data: SlopeStabilityInput):
25
+ # Convert input data into a DataFrame
26
+ input_data = pd.DataFrame([data.dict()])
27
+
28
+ # Make prediction using the model
29
+ prediction = model.predict(input_data)
30
+
31
+ return {"Factor of Safety": float(prediction[0])}