Pashtetuum commited on
Commit
7228374
1 Parent(s): e5e522f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ from tensorflow.keras.models import load_model
3
  from tensorflow.keras.preprocessing.image import load_img, img_to_array
4
  import autokeras as ak
5
  #Загружаем модель
6
- loaded_model = load_model('best_model', custom_objects=ak.CUSTOM_OBJECTS)
7
  # Загружаем label_encoder
8
  with open('label_encoder.pkl', 'rb') as le_file:
9
  label_encoder = pickle.load(le_file)
 
3
  from tensorflow.keras.preprocessing.image import load_img, img_to_array
4
  import autokeras as ak
5
  #Загружаем модель
6
+ loaded_model = load_model('best_model.h5', custom_objects=ak.CUSTOM_OBJECTS)
7
  # Загружаем label_encoder
8
  with open('label_encoder.pkl', 'rb') as le_file:
9
  label_encoder = pickle.load(le_file)