mjolnir1122 commited on
Commit
2978c04
·
verified ·
1 Parent(s): 13a15aa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -0
app.py CHANGED
@@ -5,6 +5,15 @@ import numpy as np
5
  from skimage.transform import resize
6
  from PIL import Image
7
 
 
 
 
 
 
 
 
 
 
8
  # Load the trained KNN model
9
  with open("knn_model.pkl", "rb") as f:
10
  knn_model = pickle.load(f)
 
5
  from skimage.transform import resize
6
  from PIL import Image
7
 
8
+ import pickle
9
+
10
+ try:
11
+ with open("knn_model.pkl", "rb") as f:
12
+ knn_model = pickle.load(f)
13
+ print("Model loaded successfully!")
14
+ except Exception as e:
15
+ print(f"Error loading model: {e}")
16
+
17
  # Load the trained KNN model
18
  with open("knn_model.pkl", "rb") as f:
19
  knn_model = pickle.load(f)