Update app.py

app.py

@@ -95,7 +95,7 @@ def main():
         st.markdown('<p style="text-align: center;"><label>Image : </label></p>',unsafe_allow_html=True)
         st.image(image,width=500)
     if st.button("Predict"):
-        x = pred(image)
+        x,optic,macula = 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)