Hugging Face's logo

Spaces:
Pashtetuum
/
MetallClassificationZip
Sleeping

App Files Community
Pashtetuum commited on
Commit
aa9d0f7
•
1 Parent(s): d3c240f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ import pickle
6
  import gradio as gr
7
  import cv2
8
  #Загружаем модель
9
- loaded_model = load_model('best_model.h5', custom_objects=ak.CUSTOM_OBJECTS)
10
  # Загружаем label_encoder
11
  with open('label_encoder.pkl', 'rb') as le_file:
12
  label_encoder = pickle.load(le_file)
 
6
  import gradio as gr
7
  import cv2
8
  #Загружаем модель
9
+ loaded_model = load_model('best_model', custom_objects=ak.CUSTOM_OBJECTS)
10
  # Загружаем label_encoder
11
  with open('label_encoder.pkl', 'rb') as le_file:
12
  label_encoder = pickle.load(le_file)