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 vit-base-patch16-224-in21k by Google, 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()