Spaces:
Runtime error
Runtime error
| import streamlit as st | |
| from PIL import Image | |
| from models import predict | |
| import numpy | |
| def load_image(image_file): | |
| img = Image.open(image_file) | |
| return img | |
| st.title("Cataract Image Classification") | |
| st.header('Enter the fundus image') | |
| st.subheader("KNN Model") | |
| image_file = st.file_uploader("Upload Images", type=["png", "jpg", "jpeg"]) | |
| if image_file is not None: | |
| img = load_image(image_file) | |
| st.image(img, width=250) | |
| open_cv_image = numpy.array(img) | |
| label, prob = predict(open_cv_image) | |
| st.write(f"Label : {label}") | |
| st.write(f"Probability: {prob}") | |