HumzaAli's picture
Update app.py
b74900f verified
raw
history blame contribute delete
536 Bytes
from skimage import io
import base64
from tensorflow.keras.models import load_model
import numpy as np
import gradio
import moleimages
model = load_model("mymodel-2.h5")
def predict(input):
mimg = moleimages.MoleImages()
X = mimg.load_image(input)
y_pred = model.predict(X)
return {"Benign": float(y_pred[0][0]), "Malignant": float(1-y_pred[0][0])}
io = gradio.Interface(fn=predict, inputs='image', outputs='label', analytics_enabled=True,
title="Skin Cancer Detection & Classification", description="")
io.launch()