dewifaj's picture
Update app.py
2bc34ad verified
raw
history blame
814 Bytes
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",
examples=[example_image_paths],
title="Alzheimer Recognition from MRI")
iface.launch()