Update app.py

app.py

@@ -78,7 +78,7 @@ def pred(img):
             # plt.imsave(fn,image_np)
             # plt.imshow(image_np)
             # plt.imsave(fn,image_np)
-    return(image_np)
+    return(image_np,scores[0][0]*100,scores[0][1]*100)
 
 
 
@@ -98,6 +98,8 @@ def main():
         x = pred(image)
         st.markdown('<p style="text-align: center;"><label>Prediction : </label></p>',unsafe_allow_html=True)
         st.image(x,width=900)
+        st.write('Optic Disease Score',optic)
+        st.write('Macula Score',macula)
         # result =''
         # st.success('The output is {}'.format(result))
 if __name__ == '__main__': #