File size: 836 Bytes
58ad867 b5d516f 58ad867 0cabc87 58ad867 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
import gradio as gr
from tensorflow.keras.preprocessing import image
from tensorflow.keras.models import load_model
import numpy as np
def predict(image_file):
width, height = 224, 224
img = image.img_to_array(image_file)
img = np.expand_dims(img, axis=0)
img = img / 255
model = load_model("vgg16_model.h5")
result = model.predict(img)
if result[0][0] >= 0.5:
prediction = "Malignant"
else:
prediction = "Benign"
return prediction
iface = gr.Interface(
fn=predict,
inputs=gr.Image(label="Upload Image"),
outputs=gr.Textbox(label="Predicted Results"),
title="Skin Cancer Prediction",
description="Upload an image containing skin lesion to predict if it is malignant or benign.",
theme="huggingface",
# allow_flagging=False
)
iface.launch(share=True)
|