File size: 1,139 Bytes
b940604
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
7729223
b38b039
b940604
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
from transformers import pipeline
import gradio as gr
import os 

model = pipeline(model = "AhmadHakami/alzheimer-image-classification-google-vit-base-patch16",
                 task  = "image-classification")

model.model.config.id2label = {
0: 'خفيف الخرف (Mild Demented)',
1: 'متوسط الخرف (Moderate Demented)',
2: 'غير مصاب بالخرف (Non Demented)',
3: 'خفيف جداً الخرف (Very Mild Demented)'
}

examples = []
for image in os.listdir("examples"):
    examples.append(f'examples//{image}')


gr.Interface.from_pipeline(model, 
                           title="Tbyan - تِــبْيان Alzheimer MRI Classification",
                           description="This model fine-tuned using [Google Base ViT 16×16 Pitch](hf.co/google/vit-base-patch16-224-in21k), and trained on credible MRI data from Open Access Series of Imaging Studies (OASIS) and the data annotated by experts, it aims to expedite patient results, developed by Tbyan تبيان Team at AI course by `Misk Foundation & Samsung Innovation Campus`, \n **Try it now:**", 
                           examples = examples).launch()