ankitt6174 commited on
Commit
6c58f4c
·
1 Parent(s): 96e7aa9

Error solved

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -4,8 +4,12 @@ import joblib
4
 
5
  app = FastAPI()
6
 
7
- with open("model.pkl") as mdl:
8
- model = joblib.load(mdl)
 
 
 
 
9
 
10
  class InputData(BaseModel):
11
  sepal_length: float
 
4
 
5
  app = FastAPI()
6
 
7
+ try:
8
+ model = joblib.load("model.pkl")
9
+ print("Model loaded successfully.")
10
+ except Exception as e:
11
+ print(f"Error loading model: {e}")
12
+ model = None
13
 
14
  class InputData(BaseModel):
15
  sepal_length: float