deveix commited on
Commit
2fab7e7
·
1 Parent(s): aec22dd

add new cnn model

Browse files
Files changed (1) hide show
  1. app/main.py +2 -2
app/main.py CHANGED
@@ -286,8 +286,8 @@ def generate_spectrogram(audio):
286
 
287
  return S_dB_norm
288
 
289
- cnn_model = load_model('app/cnn.h5')
290
- cnn_label_encoder = joblib.load('app/cnn_label_encoder.pkl')
291
 
292
  @app.post("/cnn")
293
  async def handle_cnn(file: UploadFile = File(...)):
 
286
 
287
  return S_dB_norm
288
 
289
+ cnn_model = load_model('app/apr23.h5')
290
+ cnn_label_encoder = joblib.load('app/apr23_label.pkl')
291
 
292
  @app.post("/cnn")
293
  async def handle_cnn(file: UploadFile = File(...)):