Spaces:
Runtime error
Runtime error
| __author__ = "Baishali Dutta" | |
| __copyright__ = "Copyright (C) 2021 Baishali Dutta" | |
| __license__ = "Apache License 2.0" | |
| __version__ = "0.1" | |
| # ------------------------------------------------------------------------- | |
| # Importing the libraries | |
| # ------------------------------------------------------------------------- | |
| import gradio as gr | |
| import numpy as np | |
| from tensorflow.keras.models import load_model | |
| from tensorflow.keras.preprocessing import image | |
| # ------------------------------------------------------------------------- | |
| # Configurations | |
| # ------------------------------------------------------------------------- | |
| MODEL_LOC = 'pneumonia_detection_cnn_model.h5' | |
| # load the trained CNN model | |
| cnn_model = load_model(MODEL_LOC) | |
| def make_prediction(test_image): | |
| test_image = test_image.name | |
| test_image = image.load_img(test_image, target_size=(224, 224)) | |
| test_image = image.img_to_array(test_image) / 255. | |
| test_image = np.expand_dims(test_image, axis=0) | |
| result = cnn_model.predict(test_image) | |
| return {"Normal": str(result[0][0]), "Pneumonia": str(result[0][1])} | |
| image_input = gr.inputs.Image(type="file") | |
| title = "Pneumonia Detection" | |
| description = "This application uses a Convolutional Neural Network (CNN) model to predict whether a chosen X-ray shows if " \ | |
| "the person has pneumonia disease or not. To check the model prediction, here are the true labels of the " \ | |
| "provided examples below: the first 4 images belong to normal whereas the last 4 images are of pneumonia " \ | |
| "category. More specifically, the 5th and 6th images are viral pneumonia infection in nature whereas " \ | |
| "the last 2 images are bacterial infection in nature." | |
| gr.Interface(fn=make_prediction, | |
| inputs=image_input, | |
| outputs="label", | |
| examples=[["image1_normal.jpeg"], | |
| ["image2_normal.jpeg"], | |
| ["image3_normal.jpeg"], | |
| ["image4_normal.jpeg"], | |
| ["image1_pneumonia_virus.jpeg"], | |
| ["image2_pneumonia_virus.jpeg"], | |
| ["image1_pneumonia_bacteria.jpeg"], | |
| ["image2_pneumonia_bacteria.jpeg"]], | |
| title=title, | |
| description=description, | |
| article="http://raw.githubusercontent.com/baishalidutta/Pneumonia-Detection/gradio/README.md") \ | |
| .launch(share=True) | |