Spaces:

yxmauw
/

eye-disease-clf-app

Running

yxmauw commited on Oct 15, 2022

Commit

bb7cb26

•

1 Parent(s): 6eb2355

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -20,7 +20,7 @@ new_img = st.file_uploader('Please upload your retinal image in .png or .jpeg/.j
 def predict_upload():
   result = predict(new_img) # result is a probabilities array
-  classes = ['cataract', 'diabetic_retinopathy', 'glaucoma', 'normal']
   max_result = (np.max(result, axis=-1)) * 100 # max probability
   pred_prob = np.format_float_positional(max_result, precision=2) # format probability
   pred_class = classes[(np.argmax(result, axis=-1)[0])] # string

 def predict_upload():
   result = predict(new_img) # result is a probabilities array
+  classes = ['cataract', 'diabetic retinopathy', 'glaucoma', 'normal']
   max_result = (np.max(result, axis=-1)) * 100 # max probability
   pred_prob = np.format_float_positional(max_result, precision=2) # format probability
   pred_class = classes[(np.argmax(result, axis=-1)[0])] # string