Update app.py

app.py

@@ -5,7 +5,7 @@ from pydantic import BaseModel
 import joblib
 import numpy as np
 from sklearn.datasets import load_iris
-
+import os
 # Load the trained model
 model = joblib.load("iris_model.pkl")