app.py

import gradio as gr
from transformers import pipeline

pipe = pipeline(task="image-classification", 
                model="cvetanovskaa/vit-base-patch16-224-in21k-gtsrb-tuned")
                
gr.Interface.from_pipeline(pipe, 
                           title="GTSRB Image Classification",
                           description="Traffic Sign Image Classification",
                           examples=['no_passing.png', 'obstacle_on_right.png'],
                           article = "Author: <a href=\"https://huggingface.co/cvetanovskaa\">Aleksandra Cvetanovska</a>",
                           ).launch()