keremoktay1 commited on
Commit
6d25cff
1 Parent(s): 0a73b81

Upload Gradio.py

Browse files
Files changed (1) hide show
  1. Gradio.py +9 -9
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(model_directory + 'dt2.pkl', 'rb')),
13
- "3": pickle.load(open(model_directory + 'dt3.pkl', 'rb')),
14
- "5": pickle.load(open(model_directory + 'dt5.pkl', 'rb'))
15
  }
16
  knn_models = {
17
- "1": pickle.load(open(model_directory + 'knn1.pkl', 'rb')),
18
- "3": pickle.load(open(model_directory + 'knn3.pkl', 'rb')),
19
- "5": pickle.load(open(model_directory + 'knn5.pkl', 'rb'))
20
  }
21
 
22
  # Load the neural network models
23
  nn_models = {
24
- "1": load_model(model_directory + 'cnn_1layer.h5'),
25
- "2": load_model(model_directory + 'cnn_2layer.h5'),
26
- "3": load_model(model_directory + 'cnn_3layer.h5')
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):