BlueSR commited on
Commit
99be414
·
verified ·
1 Parent(s): 66609d0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ def load_cnn_model():
40
  # Load the Random Forest model
41
  def load_rf_model():
42
  with open("rf_model.pkl", "rb") as f:
43
- return pickle.load(f)
44
 
45
  # Preprocess image for CNN
46
  def preprocess_image_cnn(image):
 
40
  # Load the Random Forest model
41
  def load_rf_model():
42
  with open("rf_model.pkl", "rb") as f:
43
+ return joblib.load(f)
44
 
45
  # Preprocess image for CNN
46
  def preprocess_image_cnn(image):