sensei-ml's picture
Update app.py
106c722 verified
raw
history blame
542 Bytes
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()