Spaces:

scholar-2001
/

Image_Classify

Sleeping

scholar-2001 commited on Jul 21, 2023

Commit

a172fdb

1 Parent(s): 8b20a6c

Final Changes2.0

Files changed (1) hide show

app.py CHANGED Viewed

@@ -22,7 +22,7 @@ if upload is not None:
     c1.image(im)
     loaded_ensemble_model = load_model('ensemble_model.h5')
-    pred = loaded_ensemble_model.predict(img_array)
     labels = {0:'MildDemented',1:'ModerateDemented',2:'NonDemented',3:'VeryMildDemented'}
     c2.header('Output')
     c2.subheader('Predicted class :')

     c1.image(im)
     loaded_ensemble_model = load_model('ensemble_model.h5')
+    pred = loaded_ensemble_model.predict([img_array,img_array,img_array])
     labels = {0:'MildDemented',1:'ModerateDemented',2:'NonDemented',3:'VeryMildDemented'}
     c2.header('Output')
     c2.subheader('Predicted class :')