Spaces:
Sleeping
Sleeping
File size: 813 Bytes
9178703 9b31f7b 6616b1a 9b31f7b 9178703 9b31f7b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
from transformers import pipeline
import gradio as gr
def alz_mri_classification(image):
classifier = pipeline("image-classification", model="dewifaj/alzheimer_mri_classification")
result = classifier(image)
# extract the highest score
prediction = result[0]
score = prediction['score']
label = prediction['label']
return {"score": score, "label": label}
example_image_paths = ["Very_Mild_Demented.png", "Mild_Demented.png", "Moderate_Demented.png", "Non_Demented.png"]
image_input = gr.Image(type="pil", label="Upload Image")
iface = gr.Interface(fn=alz_mri_classification,
inputs=image_input,
outputs="json",
example = example_image_paths,
title="Alzheimer Recognition from MRI")
iface.launch()
|