from model import model | |
import gradio as gr | |
import numpy as np | |
def classify_image(image): | |
probabilities = model.predict(image) # Call prediction function | |
return probabilities | |
demo = gr.Interface( | |
fn=classify_image, | |
inputs=gr.Image(), | |
outputs=gr.Label(num_top_classes=4), | |
title="Brain tumor image MRI classification", | |
description="CNN Keras model to classify brain MRI images in 4 categories: Glioma, Meningioma, Pituitary and No tumor. AI model and interface made by Eng. Michael Leonardo Aguas",) | |
demo.launch() |