app.py

import gradio as gr
from transformers import pipeline, AutoFeatureExtractor, AutoModelForImageClassification

extractor = AutoFeatureExtractor.from_pretrained("nateraw/vit-age-classifier")

model = AutoModelForImageClassification.from_pretrained("nateraw/vit-age-classifier")

pipe = pipeline(task="image-classification", 
                model=model,
                )

gr.Interface.load(pipe, 
                    title="age class",
                    description="Add description",
                    ).launch()