app.py

import gradio as gr

from datasets import load_dataset

dataset = load_dataset("cifar100")
dataset = load_dataset("fashion_mnist")

gr.Interface.load("models/google/vit-base-patch16-224").launch()
gr.Interface.load("models/microsoft/resnet-18").launch()