scholar-2001 commited on
Commit
2ed07e2
·
1 Parent(s): 535ab94

New improvements

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. ensemble_model.h5 +2 -2
app.py CHANGED
@@ -22,7 +22,7 @@ if upload is not None:
22
  c1.image(im)
23
 
24
  loaded_ensemble_model = load_model('ensemble_model.h5')
25
- pred = loaded_ensemble_model.predict([img_array, img_array])
26
  labels = {0:'MildDemented',1:'ModerateDemented',2:'NonDemented',3:'VeryMildDemented'}
27
  output_array = pred
28
  normalized_array = output_array / np.sum(output_array)
 
22
  c1.image(im)
23
 
24
  loaded_ensemble_model = load_model('ensemble_model.h5')
25
+ pred = loaded_ensemble_model.predict(img_array)
26
  labels = {0:'MildDemented',1:'ModerateDemented',2:'NonDemented',3:'VeryMildDemented'}
27
  output_array = pred
28
  normalized_array = output_array / np.sum(output_array)
ensemble_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b0b4b9414ac087b28610def93474a36f5ec9b5f71c8370fb3e5c6aa3d9c3e77b
3
- size 332868328
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1cbc5117de51702d5f18d0ffee0ecd2c6701ff7956d57be91fb94e403625dd40
3
+ size 252826456