Spaces:
Sleeping
Sleeping
keremoktay1
commited on
Commit
•
6d25cff
1
Parent(s):
0a73b81
Upload Gradio.py
Browse files
Gradio.py
CHANGED
@@ -9,21 +9,21 @@ model_directory = '/Users/keremoktay/Desktop/Gradio/'
|
|
9 |
|
10 |
# Load the decision tree and KNN models
|
11 |
dt_models = {
|
12 |
-
"2": pickle.load(open(
|
13 |
-
"3": pickle.load(open(
|
14 |
-
"5": pickle.load(open(
|
15 |
}
|
16 |
knn_models = {
|
17 |
-
"1": pickle.load(open(
|
18 |
-
"3": pickle.load(open(
|
19 |
-
"5": pickle.load(open(
|
20 |
}
|
21 |
|
22 |
# Load the neural network models
|
23 |
nn_models = {
|
24 |
-
"1": load_model(
|
25 |
-
"2": load_model(
|
26 |
-
"3": load_model(
|
27 |
}
|
28 |
|
29 |
def preprocess_image_for_ml(image_path):
|
|
|
9 |
|
10 |
# Load the decision tree and KNN models
|
11 |
dt_models = {
|
12 |
+
"2": pickle.load(open('dt2.pkl', 'rb')),
|
13 |
+
"3": pickle.load(open('dt3.pkl', 'rb')),
|
14 |
+
"5": pickle.load(open('dt5.pkl', 'rb'))
|
15 |
}
|
16 |
knn_models = {
|
17 |
+
"1": pickle.load(open('knn1.pkl', 'rb')),
|
18 |
+
"3": pickle.load(open('knn3.pkl', 'rb')),
|
19 |
+
"5": pickle.load(open('knn5.pkl', 'rb'))
|
20 |
}
|
21 |
|
22 |
# Load the neural network models
|
23 |
nn_models = {
|
24 |
+
"1": load_model('cnn_1layer.h5'),
|
25 |
+
"2": load_model('cnn_2layer.h5'),
|
26 |
+
"3": load_model('cnn_3layer.h5')
|
27 |
}
|
28 |
|
29 |
def preprocess_image_for_ml(image_path):
|