Spaces:
Sleeping
Sleeping
| from transformers import pipeline | |
| import gradio as gr | |
| def alz_mri_classification(image): | |
| classifier = pipeline("image-classification", model="dewifaj/alzheimer_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() |