itachi-ai's picture
Upload 6 files
437e081 verified
raw
history blame
798 Bytes
import gradio as gr
from predict import makepredictions
with gr.Blocks() as demo:
gr.Markdown("Brain Tumor Image classification")
gr.Markdown("We detect 4 Types of Tumor with 99.12% Accuracy:")
gr.Markdown("Glioma | Meningioma | No Tumor | Pituitary")
image_input = gr.Image(label='Tumor Image', type='filepath', height=480)
with gr.Row():
image_button = gr.Button("Predict")
y_pred = output = gr.Textbox(label="Tumor Type:")
image_button.click(makepredictions, inputs=image_input, outputs=y_pred)
gr.Markdown(
"""
This model is trained on the Kaggle dataset:
- Only for the educational purpose
- Play with that and Have fun.
""")
if __name__ == "__main__":
demo.launch()