|
from model import model |
|
import gradio as gr |
|
import numpy as np |
|
|
|
def classify_image(image): |
|
probabilities = model.predict(image) |
|
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="Convolutional neural network proprietary model made in Keras to classify brain MRI images in 4 categories: Glioma, Meningioma, Pituitary and No tumor. AI model and interface built by Eng. Michael Leonardo Aguas.",) |
|
demo.launch() |
